Update app.py

app.py

@@ -1,14 +1,9 @@
 import gradio as gr
-
-# Set the theme
-gr.Interface.load_theme(gr.themes.Soft())
+from mistral_client import InferenceClient
 
 # Initialize the InferenceClient
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-# The base of this project is made by ehristoforu
-# This code was mostly written by vericudebuget and gpt-4
-
 def format_prompt(message, history):
     prompt = "<s>"
     for user_prompt, bot_response in history:
@@ -39,13 +34,6 @@ def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=904
         yield output
     return output
 
-    for response in stream:
-        output += response.token.text
-        if "http" in output:  # assuming the AI writes a direct image link in its response
-            yield {"image": output}  # Gradio will display the image
-        else:
-            yield output
-
 additional_inputs = [
     gr.Textbox(label="System Prompt", max_lines=1, interactive=True),
     gr.Slider(label="Temperature", value=0.9, minimum=0.0, maximum=1.0, step=0.05, interactive=True, info="Higher values produce more diverse outputs"),
@@ -62,4 +50,4 @@ gr.Interface(
     description="Remember! The AI might give incorrect information about people, locations, history, etc...",
     concurrency_limit=20,
     theme=gr.themes.Soft()  # Set the theme here
-).launch(show_api=False,)
+).launch(show_api=False,)