Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ class BasicAgent:
|
|
37 |
|
38 |
# Initialize the model with TransformersModel
|
39 |
model = TransformersModel(
|
40 |
-
model_id="
|
41 |
max_new_tokens=4096,
|
42 |
device_map="auto"
|
43 |
)
|
|
|
37 |
|
38 |
# Initialize the model with TransformersModel
|
39 |
model = TransformersModel(
|
40 |
+
model_id="gpt2",
|
41 |
max_new_tokens=4096,
|
42 |
device_map="auto"
|
43 |
)
|