Update app.py

app.py

@@ -4,7 +4,14 @@ import io
 import random
 import os
 from PIL import Image
+from huggingface_hub import InferenceApi, InferenceClient
 
+client = InferenceClient()
+
+models = client.list_deployed_models()
+list_models = models["text-to-image"]
+
+'''
 list_models = [
     "SDXL-1.0",
     "SD-1.5",
@@ -15,10 +22,12 @@ list_models = [
     "Dalle-3-XL",
     "Midjourney-V4-XL",
 ]
+'''
 
 def generate_txt2img(current_model, prompt, is_negative=False, image_style="None style", steps=50, cfg_scale=7,
-                     seed=None):
+                     seed=None, api_dict = api_dict):
 
+    '''
     if current_model == "SD-1.5":
         API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
     elif current_model == "SDXL-1.0":
@@ -34,11 +43,14 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
     elif current_model == "Dalle-3-XL":
         API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
     elif current_model == "Midjourney-V4-XL":
-        API_URL = "https://api-inference.huggingface.co/models/openskyml/midjourney-v4-xl"    
+        API_URL = "https://api-inference.huggingface.co/models/openskyml/midjourney-v4-xl"
+    '''
 
     API_TOKEN = os.environ.get("HF_READ_TOKEN")
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
+    api = InferenceApi(current_model)
+    api.headers = headers
 
     if image_style == "None style":
         payload = {
@@ -73,9 +85,13 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
             "seed": seed if seed is not None else random.randint(-1, 2147483647)
         }
 
-    image_bytes = requests.post(API_URL, headers=headers, json=payload).content
+    #image_bytes = requests.post(API_URL, headers=headers, json=payload).content
+    image = model_api(data = payload)
+    return image
+    '''
     image = Image.open(io.BytesIO(image_bytes))
     return image
+    '''
 
 
 css = """