Sandiago21 commited on
Commit
f52d6bf
·
verified ·
1 Parent(s): 3893807

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -29,7 +29,8 @@ class BasicAgent:
29
 
30
 
31
  model = TransformersModel(
32
- model_id="microsoft/phi-2", # ✅ small, reliable
 
33
  max_new_tokens=1024,
34
  temperature=0.0,
35
  )
 
29
 
30
 
31
  model = TransformersModel(
32
+ # model_id="microsoft/phi-2", # ✅ small, reliable
33
+ model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
34
  max_new_tokens=1024,
35
  temperature=0.0,
36
  )