Final_Assignment_Template

Runtime error

LamiaYT commited on Jun 26

Commit

629854c

1 Parent(s): 03ca047

Deploy GAIA agent

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,7 +60,7 @@ class EnhancedModel:
     def __init__(self):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         models_to_try = [
-            "microsoft/DialoGPT-medium",
             "distilgpt2",
             "gpt2"
         ]

     def __init__(self):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         models_to_try = [
+            "google/flan-t5-base",
             "distilgpt2",
             "gpt2"
         ]