Update app.py

app.py

@@ -196,6 +196,9 @@ def load_user_guide():
     with open('user_guide.txt', 'r') as file:
         return file.read()
 
+# Combine both model dictionaries
+all_models = {**models, **quantized_models}
+
 # Create the chat interface using Gradio Blocks
 with gr.Blocks() as demo:
     with gr.Tabs():
@@ -204,7 +207,7 @@ with gr.Blocks() as demo:
 
             gr.Markdown("## Context")
             with gr.Group():
-                model_dropdown = gr.Dropdown(choices=list(models.keys()), label="Select Model", value=selected_model)
+                model_dropdown = gr.Dropdown(choices=list(all_models.keys()), label="Select Model", value=default_model_name)
                 user_dropdown = gr.Dropdown(choices=user_names, label="Select User Name")
                 initial_story = stories[0]["title"] if stories else None
                 story_dropdown = gr.Dropdown(choices=[story["title"] for story in stories], label="Select Story", value=initial_story)