Update app.py

app.py

@@ -1,35 +1,20 @@
-import os
+
 import gradio as gr
 import requests
 from PIL import Image
 import io
 import numpy as np
 
-api_key = 'sk-CED85fi0ZhUDMWg4GvFQ5k53o7yoL7WOaPyPQcb8zPi7eDGi' # your Stability AI API key
+api_key = "sk-CED85fi0ZhUDMWg4GvFQ5k53o7yoL7WOaPyPQcb8zPi7eDGi"
 
-def stable_diffusion(user_input):
-    prompt = user_input
-    print("Prompt:", prompt)
-    
-    # Create the Stable Difusion image using the Stability AI API
+def dream_studio(user_input):
+    # Create the Dream Studio image
     response = requests.post(
-        'https://api.stability.ai/v1/image/generate',
+        'https://beta.dreamstudio.ai/api/creations/images/generate',
         headers={'Authorization': f'Bearer {api_key}'},
         json={
-            'text': prompt,
-            'model': 'sd-c2',
-            'steps': 200,
-            'size': 512,
-            'start_size': 64,
-            'start_scale': 1,
-            'end_scale': 0.25,
-            'deterministic': False,
-            'clip': False,
-            'top_k': 0,
-            'top_p': 1,
-            'temperature': 1,
-            'seed': None,
-            'noise_seed': None,
+            'model': 'biggan-deep-256',
+            'text': user_input
         }
     )
     response.raise_for_status()
@@ -41,5 +26,5 @@ def stable_diffusion(user_input):
     
     return img_arr
 
-iface = gr.Interface(fn=stable_diffusion, inputs="text", outputs="image", title="bhAI (text to image)")
+iface = gr.Interface(fn=dream_studio, inputs="text", outputs="image", title="bhAI (text to image)")
 iface.launch()