d13
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
| 16 |
class BasicAgent:
|
| 17 |
def __init__(self):
|
| 18 |
self.token = os.getenv("token")
|
| 19 |
-
self.model_id = "DeepPavlov/
|
| 20 |
self.client = InferenceClient(model=self.model_id, token=self.token)
|
| 21 |
|
| 22 |
def __call__(self, question: str) -> str:
|
|
|
|
| 16 |
class BasicAgent:
|
| 17 |
def __init__(self):
|
| 18 |
self.token = os.getenv("token")
|
| 19 |
+
self.model_id = "DeepPavlov/distilrubert-tiny-cased-conversational"
|
| 20 |
self.client = InferenceClient(model=self.model_id, token=self.token)
|
| 21 |
|
| 22 |
def __call__(self, question: str) -> str:
|