Trial and error.
Browse files
app.py
CHANGED
@@ -16,55 +16,21 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
16 |
class BasicAgent:
|
17 |
def __init__(self):
|
18 |
print("BasicAgent initialized.")
|
19 |
-
#
|
20 |
-
|
21 |
-
self.model_name = "https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud"
|
22 |
self.hf_token = os.getenv("HF_TOKEN") # Get token from environment if available
|
23 |
|
24 |
-
# List of fallback models (free models that don't require authentication)
|
25 |
-
self.fallback_models = []
|
26 |
-
|
27 |
try:
|
28 |
-
print(f"
|
29 |
self.hf_client = InferenceClient(
|
30 |
model=self.model_name,
|
31 |
token=self.hf_token
|
32 |
)
|
33 |
-
# Test the client with a simple prompt to verify it works
|
34 |
-
test_response = self.hf_client.text_generation(
|
35 |
-
prompt="Hello, this is a test.",
|
36 |
-
max_new_tokens=10
|
37 |
-
)
|
38 |
print(f"Model initialized successfully: {self.model_name}")
|
39 |
-
|
40 |
except Exception as e:
|
41 |
-
print(f"Error initializing
|
42 |
self.hf_client = None
|
43 |
-
|
44 |
-
# Try fallback models
|
45 |
-
for fallback_model in self.fallback_models:
|
46 |
-
try:
|
47 |
-
print(f"Attempting fallback model: {fallback_model}")
|
48 |
-
self.hf_client = InferenceClient(
|
49 |
-
model=fallback_model,
|
50 |
-
token=self.hf_token
|
51 |
-
)
|
52 |
-
# Quick test to verify the model works
|
53 |
-
test_response = self.hf_client.text_generation(
|
54 |
-
prompt="Hello, this is a test.",
|
55 |
-
max_new_tokens=5
|
56 |
-
)
|
57 |
-
print(f"Successfully initialized fallback model: {fallback_model}")
|
58 |
-
self.model_name = fallback_model
|
59 |
-
break
|
60 |
-
except Exception as fallback_error:
|
61 |
-
print(f"Fallback model failed ({fallback_model}): {fallback_error}")
|
62 |
-
self.hf_client = None
|
63 |
-
|
64 |
-
# If all models fail, we'll use a rule-based response generator
|
65 |
-
if self.hf_client is None:
|
66 |
-
print("WARNING: All models failed. Using rule-based fallback for responses.")
|
67 |
-
self.model_name = "rule-based-fallback"
|
68 |
|
69 |
def break_down_question(self, question: str) -> list:
|
70 |
"""
|
@@ -265,6 +231,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
|
|
265 |
agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
|
266 |
print(agent_code)
|
267 |
|
|
|
|
|
268 |
# 2. Fetch Questions
|
269 |
print(f"Fetching questions from: {questions_url}")
|
270 |
try:
|
|
|
16 |
class BasicAgent:
|
17 |
def __init__(self):
|
18 |
print("BasicAgent initialized.")
|
19 |
+
# Use Qwen2.5-7B-Instruct model
|
20 |
+
self.model_name = "Qwen/Qwen2.5-7B-Instruct"
|
|
|
21 |
self.hf_token = os.getenv("HF_TOKEN") # Get token from environment if available
|
22 |
|
|
|
|
|
|
|
23 |
try:
|
24 |
+
print(f"Initializing model: {self.model_name}")
|
25 |
self.hf_client = InferenceClient(
|
26 |
model=self.model_name,
|
27 |
token=self.hf_token
|
28 |
)
|
|
|
|
|
|
|
|
|
|
|
29 |
print(f"Model initialized successfully: {self.model_name}")
|
|
|
30 |
except Exception as e:
|
31 |
+
print(f"Error initializing model ({self.model_name}): {e}")
|
32 |
self.hf_client = None
|
33 |
+
print("WARNING: Model initialization failed. Agent may not function properly.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
34 |
|
35 |
def break_down_question(self, question: str) -> list:
|
36 |
"""
|
|
|
231 |
agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
|
232 |
print(agent_code)
|
233 |
|
234 |
+
return
|
235 |
+
|
236 |
# 2. Fetch Questions
|
237 |
print(f"Fetching questions from: {questions_url}")
|
238 |
try:
|