Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-
+import google.generativeai as genai
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
@@ -22,6 +22,18 @@ def respond(
 
     messages.append({"role": "user", "content": message})
     print (message,image)
+    ## for image
+    if isinstance(image, np.ndarray):
+            img = PIL.Image.fromarray(image)
+        else:
+            try:
+                img = PIL.Image.open(image)
+            except (AttributeError, IOError) as e:
+                return f"Invalid image provided. Please provide a valid image file. Error: {e}"
+    
+        # Load model
+        model = genai.GenerativeModel("gemini-pro-vision")
+    ## for image
     response = ""
 
     for message in client.chat_completion(