Update app.py

app.py

@@ -27,7 +27,8 @@ AVAILABLE_MODELS = {
     "opt-350m": "facebook/opt-350m",
     "pythia-160m": "EleutherAI/pythia-160m",
     "mistral-7b": "mistralai/Mistral-7B-Instruct-v0.3",
-    "deepseek-small": "deepseek-ai/DeepSeek-V2-Lite"
+    "deepseek-small": "deepseek-ai/DeepSeek-V2-Lite",
+    "llama": "meta-llama/Llama-3.2-1B"
 }
 
 # Initialize model and tokenizer
@@ -45,7 +46,7 @@ def load_model(model_name):
         current_model_name = model_name
 
 # Load the default model at startup
-load_model("distilgpt2")
+load_model("mistral-7b")
 
 @spaces.GPU()
 def get_next_token_predictions(text, model_name, top_k=10):