Update app.py (#29)
Browse files- Update app.py (b87963b11d6e137b9991e3b8c3c2dd95bc636951)
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 |
)
|