Update app.py
Browse files
app.py
CHANGED
|
@@ -11,11 +11,12 @@ class BasicAgent:
|
|
| 11 |
def __init__(self):
|
| 12 |
print("🔄 Loading model: flan-alpaca-base...")
|
| 13 |
self.pipeline = pipeline(
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
|
|
|
| 19 |
print("✅ Model loaded.")
|
| 20 |
|
| 21 |
def __call__(self, question: str, task_id: str = None) -> str:
|
|
|
|
| 11 |
def __init__(self):
|
| 12 |
print("🔄 Loading model: flan-alpaca-base...")
|
| 13 |
self.pipeline = pipeline(
|
| 14 |
+
"text-generation",
|
| 15 |
+
model="HuggingFaceH4/zephyr-7b-beta",
|
| 16 |
+
max_new_tokens=256,
|
| 17 |
+
temperature=0.3
|
| 18 |
+
)
|
| 19 |
+
|
| 20 |
print("✅ Model loaded.")
|
| 21 |
|
| 22 |
def __call__(self, question: str, task_id: str = None) -> str:
|