DeekshithN05 commited on
Commit
3b8d66e
·
verified ·
1 Parent(s): 06cbea3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -11,11 +11,12 @@ class BasicAgent:
11
  def __init__(self):
12
  print("🔄 Loading model: flan-alpaca-base...")
13
  self.pipeline = pipeline(
14
- "text2text-generation",
15
- model="declare-lab/flan-alpaca-base",
16
- max_new_tokens=128,
17
- temperature=0.3
18
- )
 
19
  print("✅ Model loaded.")
20
 
21
  def __call__(self, question: str, task_id: str = None) -> str:
 
11
  def __init__(self):
12
  print("🔄 Loading model: flan-alpaca-base...")
13
  self.pipeline = pipeline(
14
+ "text-generation",
15
+ model="HuggingFaceH4/zephyr-7b-beta",
16
+ max_new_tokens=256,
17
+ temperature=0.3
18
+ )
19
+
20
  print("✅ Model loaded.")
21
 
22
  def __call__(self, question: str, task_id: str = None) -> str: