Update app.py
Browse files
app.py
CHANGED
|
@@ -95,6 +95,9 @@ model = TransformersModel(
|
|
| 95 |
model_id=model_id,
|
| 96 |
max_new_tokens=256)
|
| 97 |
|
|
|
|
|
|
|
|
|
|
| 98 |
class BasicAgent:
|
| 99 |
def __init__(self):
|
| 100 |
print("BasicAgent initialized.")
|
|
|
|
| 95 |
model_id=model_id,
|
| 96 |
max_new_tokens=256)
|
| 97 |
|
| 98 |
+
model = HfApiModel(model_id,max_new_tokens=256)
|
| 99 |
+
|
| 100 |
+
|
| 101 |
class BasicAgent:
|
| 102 |
def __init__(self):
|
| 103 |
print("BasicAgent initialized.")
|