CR7CAD commited on
Commit
13ca156
·
verified ·
1 Parent(s): d746933

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ def load_models():
48
  # Load model for evaluation
49
  models['evaluator'] = pipeline(
50
  "text2text-generation",
51
- model="google-t5/t5-small",
52
  max_length=300
53
  )
54
 
 
48
  # Load model for evaluation
49
  models['evaluator'] = pipeline(
50
  "text2text-generation",
51
+ model="distilbert/distilgpt2",
52
  max_length=300
53
  )
54