Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
| 14 |
class BasicAgent:
|
| 15 |
def __init__(self):
|
| 16 |
print("Loading Hugging Face model...")
|
| 17 |
-
self.pipeline = pipeline("text-generation", model="
|
| 18 |
print("Model loaded.")
|
| 19 |
|
| 20 |
def __call__(self, question: str) -> str:
|
|
|
|
| 14 |
class BasicAgent:
|
| 15 |
def __init__(self):
|
| 16 |
print("Loading Hugging Face model...")
|
| 17 |
+
self.pipeline = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2", max_new_tokens=256)
|
| 18 |
print("Model loaded.")
|
| 19 |
|
| 20 |
def __call__(self, question: str) -> str:
|