GenAIJake commited on
Commit
3a63ef3
·
1 Parent(s): c24b0df
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -9,7 +9,13 @@ headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
9
 
10
  def generate(prompt):
11
  try:
12
- response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
 
 
 
 
 
 
13
  image = Image.open(io.BytesIO(response.content))
14
  return image
15
  except Exception as e:
 
9
 
10
  def generate(prompt):
11
  try:
12
+ payload = {
13
+ "inputs": prompt,
14
+ "parameters": {
15
+ "num_inference_steps": 5
16
+ }
17
+ }
18
+ response = requests.post(API_URL, headers=headers, json=payload)
19
  image = Image.open(io.BytesIO(response.content))
20
  return image
21
  except Exception as e: