Update app.py

app.py

@@ -23,12 +23,11 @@ import spaces
 
 # Available models
 AVAILABLE_MODELS = {
-    "distilgpt2": "distilgpt2",
     "bloomz-560m": "bigscience/bloomz-560m",
-    "gpt2-medium": "gpt2-medium",
     "opt-350m": "facebook/opt-350m",
     "pythia-160m": "EleutherAI/pythia-160m",
-    "mistral-7b": "mistralai/Mistral-7B-Instruct"
+    "mistral-7b": "mistralai/Mistral-7B-Instruct-v0.3",
+    "deepseek-small": "deepseek-ai/DeepSeek-V2-Lite"
 }
 
 # Initialize model and tokenizer