DeekshithN05 commited on
Commit
d1b855f
·
verified ·
1 Parent(s): a50a227

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,10 +14,10 @@ from transformers import pipeline
14
 
15
  class BasicAgent:
16
  def __init__(self):
17
- print("Loading flan-t5-base model...")
18
  self.pipeline = pipeline(
19
  "text2text-generation",
20
- model="google/flan-t5-base",
21
  max_new_tokens=128,
22
  temperature=0.3
23
  )
 
14
 
15
  class BasicAgent:
16
  def __init__(self):
17
+ print("Loading LaMini-T5-738M model...")
18
  self.pipeline = pipeline(
19
  "text2text-generation",
20
+ model="MBZUAI/LaMini-T5-738M",
21
  max_new_tokens=128,
22
  temperature=0.3
23
  )