Update app.py

app.py

@@ -1,11 +1,11 @@
 # app.py
 import gradio as gr
-from chat import generate_response, reset_conversation, switch_model
+from chat import generate_response_stream, reset_conversation, switch_model
 from config import MODEL_LIST, MAX_TOKENS, TEMPERATURE
 
 def chat_interface(user_input, max_tokens, temperature, model_name):
-    switch_model(model_name)  # Switch model dynamically
-    return generate_response(user_input, max_length=max_tokens, temperature=temperature)
+    switch_model(model_name)
+    return generate_response_stream(user_input, max_length=max_tokens, temperature=temperature)
 
 iface = gr.Interface(
     fn=chat_interface,
@@ -15,9 +15,9 @@ iface = gr.Interface(
         gr.Slider(minimum=0.1, maximum=1.0, step=0.1, label="Temperature", value=TEMPERATURE),
         gr.Dropdown(list(MODEL_LIST.keys()), label="Choose Model", value="ChatGpt Tune Base")
     ],
-    outputs="text",
-    title="ChatGpt Tune Multi-Model",
-    description="Chat with multiple Hugging Face/safetensor models!"
+    outputs=gr.Textbox(label="AI Response"),
+    title="ChatGpt Tune Multi-Model Streaming",
+    description="Chat with multiple Hugging Face/safetensor models with streaming responses!"
 )
 
 reset_iface = gr.Interface(