Update app.py

app.py

@@ -5,7 +5,7 @@ import PIL.Image
 
 genai.configure(api_key="AIzaSyAj-b3sO_wUguMdpXWScxKzMHxb8C5cels")
 
-def ImageChat(image):
+def ImageChat(image, prompt):
 
   # load model
   model = genai.GenerativeModel("gemini-pro-vision")
@@ -17,18 +17,15 @@ def ImageChat(image):
   else:
         img = PIL.Image.open(image)
 
-  response =  model.generate_content(["write a short story about the image", img])
+  response =  model.generate_content([prompt, img])
 
   return response.text
 
 
 app = gr.Interface(ImageChat,
-                   inputs = gr.Image(),
+                   inputs = [gr.Image(), gr.Text()],
                    outputs = gr.Text(),
-                   title = "Image-To-Story",
+                   title = "Image Chat",
                    theme = gr.themes.Soft())
 
-
 app.launch()
-
-