Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ async def load_models():
|
|
30 |
"Salesforce/codegen-350M-multi",
|
31 |
"Salesforce/codegen-2B-multi"
|
32 |
]
|
33 |
-
gpt_models = ["gpt2-medium", "gpt2-large", "gpt2", "gemma-2-9b"] + programming_models
|
34 |
|
35 |
models = []
|
36 |
for model_name in gpt_models:
|
|
|
30 |
"Salesforce/codegen-350M-multi",
|
31 |
"Salesforce/codegen-2B-multi"
|
32 |
]
|
33 |
+
gpt_models = ["gpt2-medium", "gpt2-large", "gpt2", "google/gemma-2-9b"] + programming_models
|
34 |
|
35 |
models = []
|
36 |
for model_name in gpt_models:
|