Update app.py

app.py

@@ -28,6 +28,20 @@ def image_to_base64(image_path):
         encoded_string = base64.b64encode(img.read()).decode('utf-8')
     return f"data:image/jpeg;base64,{encoded_string}"
 
+import requests
+import base64
+
+def url_to_base64(image_url):
+    # Fetch the image from the URL
+    response = requests.get(image_url)
+    if response.status_code == 200:
+        # Encode image content to Base64
+        base64_image = base64.b64encode(response.content).decode('utf-8')
+        return base64_image
+    else:
+        return ''
+
+
 @spaces.GPU(duration=90)
 def run_inference(message, history):
     ## may work