Update app.py

app.py

@@ -2,49 +2,36 @@ import gradio as gr
 import requests
 import io
 import os
+import json
 import random
 from PIL import Image
 
 def query(model_name, prompt, is_negative, steps, cfg_scale, seed, strength, width, height):
     API_URL = f"https://api-inference.huggingface.co/models/{model_name}"
-    token = os.getenv("token")
+    headers = {"Authorization": f"Bearer {os.getenv('token')}"}
     
-    if not token:
-        return "Error: Hugging Face API token is not set."
-    
-    headers = {"Authorization": f"Bearer {token}"}
-
-    negative_prompt = "low quality, bad anatomy, worst quality, blurry" if is_negative else ""
-
     payload = {
         "inputs": prompt,
+        "is_negative": is_negative,
+        "steps": steps,
+        "cfg_scale": cfg_scale,
+        "seed": seed if seed != -1 else random.randint(1, 1000000000),
+        "strength": strength,
         "parameters": {
-            "is_negative": is_negative,
-            "num_inference_steps": steps,
-            "guidance_scale": cfg_scale,
-            "seed": seed if seed != -1 else random.randint(1, 1000000000),
-            "strength": strength,
             "width": width,
             "height": height
         }
     }
-
     response = requests.post(API_URL, headers=headers, json=payload)
-
+    
     if response.status_code != 200:
-        try:
-            error_message = response.json().get("error", response.text)
-        except Exception:
-            error_message = response.text
-        return f"Error {response.status_code}: {error_message}"
-
+        return f"Error {response.status_code}: {response.text}"
+    
     try:
         image = Image.open(io.BytesIO(response.content))
-        if image.format not in ["PNG", "JPEG", "WEBP"]:
-            return "Error: Received an invalid image format."
         return image
-    except Exception as e:
-        return f"Invalid response: {str(e)}"
+    except Exception:
+        return f"Invalid response: {response.text}"
 
 demo = gr.Interface(
     fn=query,
@@ -64,4 +51,4 @@ demo = gr.Interface(
     description="テキストプロンプトとオプションを設定して画像を生成します。"
 )
 
-demo.launch()
+demo.launch()