Trial and error.
Browse files
app.py
CHANGED
|
@@ -21,21 +21,26 @@ class BasicAgent:
|
|
| 21 |
self.model_name = "Qwen/Qwen2.5-7B-Instruct"
|
| 22 |
|
| 23 |
# Load model and tokenizer
|
| 24 |
-
try:
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
except Exception as e:
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
|
| 40 |
def __call__(self, question: str) -> str:
|
| 41 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
|
@@ -204,7 +209,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
|
|
| 204 |
|
| 205 |
# --- Build Gradio Interface using Blocks ---
|
| 206 |
with gr.Blocks() as demo:
|
| 207 |
-
gr.Markdown("# Basic Agent Evaluation Runner")
|
| 208 |
gr.Markdown(
|
| 209 |
"""
|
| 210 |
**Instructions:**
|
|
|
|
| 21 |
self.model_name = "Qwen/Qwen2.5-7B-Instruct"
|
| 22 |
|
| 23 |
# Load model and tokenizer
|
| 24 |
+
#try:
|
| 25 |
+
# self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|
| 26 |
+
# self.model = AutoModelForCausalLM.from_pretrained(
|
| 27 |
+
# self.model_name,
|
| 28 |
+
# torch_dtype="auto",
|
| 29 |
+
# device_map="auto"
|
| 30 |
+
# )
|
| 31 |
+
# print(f"Successfully loaded {self.model_name}")
|
| 32 |
+
#except Exception as e:
|
| 33 |
+
# print(f"Error loading model: {e}")
|
| 34 |
+
# # Fallback to HuggingFace Inference API if local loading fails
|
| 35 |
+
# print("Falling back to InferenceClient")
|
| 36 |
+
# self.client = InferenceClient(model=self.model_name)
|
| 37 |
+
# self.tokenizer = None
|
| 38 |
+
# self.model = None
|
| 39 |
+
print("Falling back to InferenceClient")
|
| 40 |
+
self.client = InferenceClient(model=self.model_name)
|
| 41 |
+
self.tokenizer = None
|
| 42 |
+
self.model = None
|
| 43 |
+
|
| 44 |
|
| 45 |
def __call__(self, question: str) -> str:
|
| 46 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
|
|
|
| 209 |
|
| 210 |
# --- Build Gradio Interface using Blocks ---
|
| 211 |
with gr.Blocks() as demo:
|
| 212 |
+
gr.Markdown("# Basic Agent Evaluation Runner #")
|
| 213 |
gr.Markdown(
|
| 214 |
"""
|
| 215 |
**Instructions:**
|