wt002 commited on
Commit
244ae60
·
verified ·
1 Parent(s): 6c7b3e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
23
 
24
 
25
  class BasicAgent:
26
- def __init__(self, model="google/gemma-2b"): # Smaller 2B version recommended
27
  self.tokenizer = AutoTokenizer.from_pretrained(model)
28
  self.model = AutoModelForCausalLM.from_pretrained(
29
  model,
 
23
 
24
 
25
  class BasicAgent:
26
+ def __init__(self, model="mistral-7b-instruct-v0.1"): # Smaller model recommended
27
  self.tokenizer = AutoTokenizer.from_pretrained(model)
28
  self.model = AutoModelForCausalLM.from_pretrained(
29
  model,