Spaces:
Sleeping
Sleeping
File size: 1,444 Bytes
959296e c92c3bf d93dbdd 9b5b26a 0f61d7b 2764a1c 1c61c68 4c2a00d d4578f1 4db54a2 e55e264 67c8f73 7686479 ed36d14 95de6d7 9b5b26a 8f16109 95c49b7 826fb71 65b9821 6dc9c22 0f61d7b 95c49b7 0f61d7b 4999ab5 1c61c68 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import sys # Add this import to fix the error
import datetime
import requests
from checks.failed_check import create_failed_gradio_ui
from checks.endpoint_check import check_public_endpoint
import os
import pytz # Had to give it permission in Code agent
from tools.tools import load_tools
from agents.model import load_huggingface_model
from agents.prompts import load_prompts
from Gradio_UI import GradioUI
# Prechecks
# Get the URI for the endpoint
endpoint_uri = load_huggingface_model()
# Test the endpoint
response = check_public_endpoint(endpoint_uri)
if response.status is False:
create_failed_gradio_ui()
else:
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id= endpoint_uri,
custom_role_conversions=None,
)
# Load prompts
combined_prompts = load_prompts()
combined_tools = load_tools() ## from the tools folder (don't remove final answer)
agent = CodeAgent(
model=model,
tools=combined_tools,
max_steps=6,
verbosity_level=3,
grammar=None,
planning_interval=None,
name="Content Agent",
description="Evaluates whether text is polite or impolite. ",
prompt_templates=combined_prompts,
additional_authorized_imports=["pytz"]
)
GradioUI(agent).launch()
|