artificialguybr commited on
Commit
e24c40e
1 Parent(s): 5856490

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def run_lora(prompt):
23
  token = os.getenv("API_TOKEN")
24
  payload = {"inputs": f"{prompt} {trigger_word}"}
25
  image_bytes = query(payload, api_url, token)
26
- return Image.open(image_bytes).resize((300, 300)) # Resizing the image
27
 
28
  # Gradio UI
29
  print("Before Gradio Interface")
 
23
  token = os.getenv("API_TOKEN")
24
  payload = {"inputs": f"{prompt} {trigger_word}"}
25
  image_bytes = query(payload, api_url, token)
26
+ return Image.open(image_bytes)
27
 
28
  # Gradio UI
29
  print("Before Gradio Interface")