Spaces:
Sleeping
Sleeping
File size: 1,660 Bytes
959296e c92c3bf d93dbdd 9b5b26a 0f61d7b 2764a1c 1c61c68 4c2a00d d4578f1 4db54a2 e55e264 67c8f73 7686479 ed36d14 95de6d7 9b5b26a 8f16109 95c49b7 826fb71 65b9821 6dc9c22 06e3381 95c49b7 0e921e3 06e3381 6387f36 7155a63 0f61d7b 4999ab5 06e3381 a6a43c9 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 61 62 63 64 65 |
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
status_info = check_public_endpoint(endpoint_uri)
print(status_info["status"])
print(status_info["status_code"])
if status_info["status"] is False:
interface = create_failed_gradio_ui(status_info)
# Launch the UI
interface.launch(show_error=True)
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"]
)
# Launch the UI earlier or without passing through the Agent
GradioUI(agent).launch()
|