experiment

app.py

@@ -2,6 +2,7 @@ import os
 
 import gradio as gr
 import numpy as np
+import requests
 from dotenv import load_dotenv
 from huggingface_hub import InferenceClient, login
 
@@ -11,6 +12,22 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 TOKEN = None
 
+
+def download_image_locally(image_url: str, local_path: str):
+    """
+    Download an image from a URL to a local path.
+    
+    Args:
+        image_url (str):
+            The URL of the image to download.
+        local_path (str):
+            The path to save the downloaded image.
+    """
+    response = requests.get(image_url)
+    with open(local_path, "wb") as f:
+        f.write(response.content)
+    return local_path
+
 def get_token(oauth_token: gr.OAuthToken | None):
     global TOKEN
     if oauth_token and oauth_token.token: