Initial commit

app.py

@@ -1,26 +1,46 @@
 import gradio as gr
 import requests
 import os
+from PIL import Image
+import io
 
 def generate(prompt):
     API_URL = "https://api-inference.huggingface.co/models/GenAIJake/d3xt3r-l1tt13-dachshund"
-    headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
+    headers = {
+        "Authorization": f"Bearer {os.getenv('HF_TOKEN')}",
+        "Content-Type": "application/json",
+        "x-use-cache": "false",      # Disable caching for new generations
+        "x-wait-for-model": "true"   # Wait for model if it needs to load
+    }
+    
+    # Add the trigger words to the prompt
+    full_prompt = f"d3xt3r-l1tt13 Dachshund, {prompt}"
+    
     payload = {
-        "inputs": prompt,
+        "inputs": full_prompt,
         "parameters": {
             "num_inference_steps": 30,
             "guidance_scale": 7.5,
-            "negative_prompt": ["blurry, bad quality"]
+            "negative_prompt": "blurry, bad quality, worst quality, low quality",
+            "seed": None  # Random seed for each generation
         }
     }
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+    
+    try:
+        response = requests.post(API_URL, headers=headers, json=payload)
+        # Convert the binary response to a PIL Image
+        image = Image.open(io.BytesIO(response.content))
+        return image
+    except Exception as e:
+        return f"Error generating image: {str(e)}"
 
+# Create the Gradio interface
 demo = gr.Interface(
     fn=generate,
-    inputs="text",
-    outputs="image",
-    title="D3xt3r L1tt13 Dachshund Generator"
+    inputs=gr.Textbox(label="Enter your prompt"),
+    outputs=gr.Image(label="Generated Image"),
+    title="D3xt3r L1tt13 Dachshund Generator",
+    description="Generate images of a cute dachshund character. The model will automatically add 'd3xt3r-l1tt13 Dachshund' to your prompt."
 )
 
 demo.launch()

requirements.txt

@@ -3,4 +3,6 @@ diffusers
 invisible_watermark
 torch
 transformers
-xformers
+xformers
+gradio>=4.43.0
+requests