Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from gradio_client import Client, handle_file
-from IPython.display import Image, display
 
 def generate_image(prompt, image_url=None, image_file=None):
     # Initialize the client
@@ -76,4 +75,4 @@ def gradio_interface():
 
 if __name__ == "__main__":
     demo = gradio_interface()
-    demo.launch()
+    demo.launch()