DeekshithN05 commited on
Commit
749689f
·
verified ·
1 Parent(s): 2b63892

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
14
  class BasicAgent:
15
  def __init__(self):
16
  print("Loading Hugging Face model...")
17
- self.pipeline = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", max_new_tokens=256)
18
  print("Model loaded.")
19
 
20
  def __call__(self, question: str) -> str:
 
14
  class BasicAgent:
15
  def __init__(self):
16
  print("Loading Hugging Face model...")
17
+ self.pipeline = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2", max_new_tokens=256)
18
  print("Model loaded.")
19
 
20
  def __call__(self, question: str) -> str: