d16
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
| 20 |
class BasicAgent:
|
| 21 |
def __init__(self):
|
| 22 |
self.token = os.getenv("token")
|
| 23 |
-
self.model_id = "
|
| 24 |
self.client = InferenceClient(model=self.model_id, token=self.token)
|
| 25 |
|
| 26 |
def __call__(self, question: str) -> str:
|
|
|
|
| 20 |
class BasicAgent:
|
| 21 |
def __init__(self):
|
| 22 |
self.token = os.getenv("token")
|
| 23 |
+
self.model_id = "microsoft/DialoGPT-medium"
|
| 24 |
self.client = InferenceClient(model=self.model_id, token=self.token)
|
| 25 |
|
| 26 |
def __call__(self, question: str) -> str:
|