Update app.py

app.py

@@ -2,6 +2,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 import requests
+from IPython.display import display, Image
 
 model_name = "Writer/palmyra-small"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -84,17 +85,18 @@ def generate_response(prompt):
 
     generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
     
-    # Display image directly in the chat
-    image_component = gr.Image(image_url, alt="Movie Poster")
+    # Embed image directly in the response
+    display(Image(url=image_url, alt="Movie Poster"))
     
-    return f"Movie Info:\n{movie_info}\n\nGenerated Response:\n{generated_text}", image_component
+    return f"Movie Info:\n{movie_info}\n\nGenerated Response:\n{generated_text}"
 
 # Define chat function for gr.ChatInterface
 def chat_function(message, history):
-    response, image_component = generate_response(message)
+    response = generate_response(message)
     history.append([message, response])
-    return response, image_component
+    return response
 
 # Create Gradio Chat Interface
 chat_interface = gr.ChatInterface(chat_function)
-chat_interface.launch(share=True)
+chat_interface.launch(share=True)  # Added share=True to create a public link
+