martynka commited on
Commit
35e0c74
·
verified ·
1 Parent(s): 956ff4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -36,7 +36,8 @@ def upload_to_chevereto(image: Image.Image) -> str:
36
 
37
  # Generate and upload the image
38
  def generate_image_and_upload(prompt: str) -> str:
39
- img = model.submit([prompt]) # Get generated image
 
40
  return upload_to_chevereto(img) # Upload and return the URL
41
 
42
  # Define Gradio API returning the Chevereto public URL
 
36
 
37
  # Generate and upload the image
38
  def generate_image_and_upload(prompt: str) -> str:
39
+ #img = model.submit([prompt]) # Get generated image
40
+ img = model.predict(prompt)
41
  return upload_to_chevereto(img) # Upload and return the URL
42
 
43
  # Define Gradio API returning the Chevereto public URL