Update app.py

app.py

@@ -1,3 +1,24 @@
 import gradio as gr
 
-gr.load("models/mistralai/Mixtral-8x7B-Instruct-v0.1").launch()
+model = gr.load("models/mistralai/Mixtral-8x7B-Instruct-v0.1")
+
+def chat(prompt, response_type):
+    if response_type == "Short":
+        return model(prompt, max_tokens=50)
+    elif response_type == "Medium":
+        return model(prompt, max_tokens=100)
+    else:
+        return model(prompt, max_tokens=200)
+
+demo = gr.Interface(
+    fn=chat,
+    inputs=[
+        gr.Textbox(label="Prompt"),
+        gr.Radio(["Short", "Medium", "Long"], label="Response Type")
+    ],
+    outputs=gr.Textbox(label="Response"),
+    title="Mixtral-8x7B-Instruct-v0.1 Chat",
+    description="Chat with the Mixtral-8x7B-Instruct-v0.1 model."
+)
+
+demo.launch()