Update app.py

app.py

@@ -4,8 +4,7 @@ from huggingface_hub import InferenceClient
 """
 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
 """
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-
+client = InferenceClient("AI-Safeguard/Ivy-VL-llava")
 
 def respond(
     message,
@@ -14,6 +13,7 @@ def respond(
     max_tokens,
     temperature,
     top_p,
+    image=None,
 ):
     messages = [{"role": "system", "content": system_message}]
 
@@ -25,21 +25,27 @@ def respond(
 
     messages.append({"role": "user", "content": message})
 
+    payload = {
+        "messages": messages,
+        "max_tokens": max_tokens,
+        "temperature": temperature,
+        "top_p": top_p,
+    }
+
+    if image is not None:
+        payload["image"] = image
+
     response = ""
 
     for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
+        payload,
         stream=True,
-        temperature=temperature,
-        top_p=top_p,
     ):
         token = message.choices[0].delta.content
 
         response += token
         yield response
 
-
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
 """
@@ -56,9 +62,9 @@ demo = gr.ChatInterface(
             step=0.05,
             label="Top-p (nucleus sampling)",
         ),
+        gr.Image(type="filepath", label="Input Image (optional)"),
     ],
 )
 
-
 if __name__ == "__main__":
     demo.launch()