Update app.py

app.py

@@ -31,10 +31,8 @@ AVAILABLE_MODELS = {
     "bloomz-560m": "bigscience/bloomz-560m",
     "opt-350m": "facebook/opt-350m",
     "pythia-160m": "EleutherAI/pythia-160m",
-    "mistral-7b": "mistralai/Mistral-7B-v0.1",
-    "deepseek-small": "deepseek-ai/DeepSeek-V2-Lite",
-    "llama": "meta-llama/Llama-3.2-1B",
-    "eulether": "EleutherAI/gpt-j-6b"
+    "eulether": "EleutherAI/gpt-j-6b",
+    "gpt2-large": "gpt2-large",
 }
 
 # Initialize model and tokenizer
@@ -52,7 +50,7 @@ def load_model(model_name):
         current_model_name = model_name
 
 # Load the default model at startup
-load_model("mistral-7b")
+load_model("gpt2-large")
 
 @spaces.GPU()
 def get_next_token_predictions(text, model_name, top_k=10):