Update app.py

app.py

@@ -41,16 +41,6 @@ def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=904
         else:
             yield output
 
-             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
-
-             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"),