Spaces:

CR7CAD
/

ISOM5240FinalProject

Sleeping

CR7CAD commited on Mar 18

Commit

de6503f

verified ·

1 Parent(s): b69d064

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ def load_models():
         # Load T5-small model for evaluation
         models['evaluator'] = pipeline(
             "text2text-generation",
-            model="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
             max_length=200
         )

         # Load T5-small model for evaluation
         models['evaluator'] = pipeline(
             "text2text-generation",
+            model="microsoft/DialoGPT-small",
             max_length=200
         )