jiuface commited on
Commit
762ec4a
·
1 Parent(s): dd41079
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,6 +20,7 @@ from gradio_imageslider import ImageSlider
20
  import boto3
21
  from io import BytesIO
22
  from datetime import datetime
 
23
 
24
  device = "cuda"
25
  base_model_id = "SG161222/RealVisXL_V4.0"
@@ -110,7 +111,7 @@ def upload_image_to_s3(image, account_id, access_key, secret_key, bucket_name):
110
 
111
 
112
  @spaces.GPU(enable_queue=True)
113
- def process(image, image_url, prompt, n_prompt, num_steps, guidance_scale, control_strength, seed, upload_to_s3, account_id, access_key, secret_key, bucket, progress=gr.Progress()):
114
 
115
  if image_url:
116
  orginal_image = load_image(image_url)
 
20
  import boto3
21
  from io import BytesIO
22
  from datetime import datetime
23
+ import json
24
 
25
  device = "cuda"
26
  base_model_id = "SG161222/RealVisXL_V4.0"
 
111
 
112
 
113
  @spaces.GPU(enable_queue=True)
114
+ def process(image, image_url, prompt, n_prompt, num_steps, guidance_scale, control_strength, seed, upload_to_s3, account_id, access_key, secret_key, bucket):
115
 
116
  if image_url:
117
  orginal_image = load_image(image_url)