Update app.py

app.py

@@ -1,24 +1,15 @@
-'''🔧 Prerequisites
-
-Install the necessary packages:
-
-pip install gradio transformers
-
-📱 Gradio Chatbot App Code
-'''
 import gradio as gr
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 import torch
 
 # List of available premium models
 premium_models = [
-    "HuggingFaceH4/zephyr-7b-beta",
-    "K00B404/BagOClownCoders-slerp-7B",
-    "Qwen/Qwen2.5-Omni-7B",
-    "Qwen/Qwen2.5-VL-7B-Instruct",
-    "deepseek-ai/Janus-Pro-7B",
-    "meta-llama/Llama-2-7b-hf",
-    "Alibaba-NLP/gte-Qwen2-7B-instruct",
+    "Qwen/Qwen2-0.5B",
+    "Qwen/Qwen2-0.5B-Instruct",
+    "Qwen/Qwen2-0.5B-Coder",
+    "Qwen/Qwen2-1.5B",
+    "Qwen/Qwen2-1.5B-Instruct",
+    "Qwen/Qwen2-1.5B-Coder",
 ]
 
 # Dictionary to cache loaded pipelines
@@ -72,6 +63,9 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         model_choice = gr.Dropdown(label="Select Model", choices=premium_models, value=premium_models[0])
+        
+     with gr.Row():
+        model_choice = gr.Textbox(label="System Prompt", choices=premium_models, value=default_system_prompt)
     
     chatbot_ui = gr.Chatbot(type="messages")
     user_input = gr.Textbox(show_label=False, placeholder="Type your message and press Enter")
@@ -83,25 +77,3 @@ with gr.Blocks() as demo:
     clear_btn.click(lambda: ([], ""), None, [chatbot_ui, state])
 
 demo.launch()
-'''
-✅ Features:
-
-    Model selection from dropdown
-
-    Maintains chat history
-
-    Respects a system prompt
-
-    Uses text-generation pipeline
-
-🧠 Optional Upgrades:
-
-    Replace text-generation with chat-completion if models support it (like OpenChat, Mistral-instruct, etc.)
-
-    Add streaming or token-by-token response if supported
-
-    Save/load chat history
-
-    Add support for vision models (Qwen2.5-VL-7B-Instruct) using a different UI tab
-
-'''