bluenevus commited on
Commit
8bbb39c
·
verified ·
1 Parent(s): e4756d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -13
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  import google.generativeai as genai
3
  import requests
4
- import io
5
 
6
  # List of popular styles
7
  STYLES = [
@@ -55,20 +54,20 @@ def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt):
55
  "Authorization": f"Bearer {stability_api_key}"
56
  }
57
 
58
- data = {
59
- "prompt": f"{enhanced_prompt}, Style: {style}",
60
- "negative_prompt": negative_prompt,
61
- "model": "sd3.5-large-turbo",
62
- "width": "1024",
63
- "height": "1024",
64
- "num_images": "1",
65
- "steps": "30",
66
- "cfg_scale": "7.5",
67
- "seed": "0",
68
- "output_format": "png"
69
  }
70
 
71
- response = requests.post(url, headers=headers, data=data)
72
 
73
  if response.status_code == 200:
74
  return response.content
 
1
  import gradio as gr
2
  import google.generativeai as genai
3
  import requests
 
4
 
5
  # List of popular styles
6
  STYLES = [
 
54
  "Authorization": f"Bearer {stability_api_key}"
55
  }
56
 
57
+ files = {
58
+ "prompt": (None, f"{enhanced_prompt}, Style: {style}"),
59
+ "negative_prompt": (None, negative_prompt),
60
+ "model": (None, "sd3.5-large-turbo"),
61
+ "width": (None, "1024"),
62
+ "height": (None, "1024"),
63
+ "num_images": (None, "1"),
64
+ "steps": (None, "30"),
65
+ "cfg_scale": (None, "7.5"),
66
+ "seed": (None, "0"),
67
+ "output_format": (None, "png")
68
  }
69
 
70
+ response = requests.post(url, headers=headers, files=files)
71
 
72
  if response.status_code == 200:
73
  return response.content