CR7CAD commited on
Commit
2924dc7
·
verified ·
1 Parent(s): c9f8450

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def load_models():
37
  models['summarizer'] = pipeline("summarization", model="facebook/bart-large-cnn", max_length=130)
38
 
39
  # Load Phi-4 model for evaluation
40
- models['evaluator'] = pipeline("text-generation", model="microsoft/Phi-4-mini-instruct", max_new_tokens=150)
41
 
42
  return models
43
 
 
37
  models['summarizer'] = pipeline("summarization", model="facebook/bart-large-cnn", max_length=130)
38
 
39
  # Load Phi-4 model for evaluation
40
+ models['evaluator'] = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", max_new_tokens=150)
41
 
42
  return models
43