ford442 commited on
Commit
400f932
·
verified ·
1 Parent(s): 3b40ad0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -277,10 +277,10 @@ def generate_30(
277
  sd_image_path = f"rv50_B_{seed}.png"
278
  rv_image.save(sd_image_path,optimize=False,compress_level=0)
279
  upload_to_ftp(sd_image_path)
280
- image_paths = save_image(rv_image)
281
  torch.cuda.empty_cache()
282
  gc.collect()
283
- return image_paths, seed
284
 
285
  @spaces.GPU(duration=60)
286
  def generate_60(
@@ -334,10 +334,10 @@ def generate_60(
334
  sd_image_path = f"rv50_B_{seed}.png"
335
  rv_image.save(sd_image_path,optimize=False,compress_level=0)
336
  upload_to_ftp(sd_image_path)
337
- image_paths = save_image(rv_image)
338
  torch.cuda.empty_cache()
339
  gc.collect()
340
- return image_paths, seed
341
 
342
  @spaces.GPU(duration=90)
343
  def generate_90(
@@ -391,10 +391,10 @@ def generate_90(
391
  sd_image_path = f"rv50_B_{seed}.png"
392
  rv_image.save(sd_image_path,optimize=False,compress_level=0)
393
  upload_to_ftp(sd_image_path)
394
- image_paths = save_image(rv_image)
395
  torch.cuda.empty_cache()
396
  gc.collect()
397
- return image_paths, seed
398
 
399
  def load_predefined_images1():
400
  predefined_images1 = [
 
277
  sd_image_path = f"rv50_B_{seed}.png"
278
  rv_image.save(sd_image_path,optimize=False,compress_level=0)
279
  upload_to_ftp(sd_image_path)
280
+ #image_paths = save_image(rv_image)
281
  torch.cuda.empty_cache()
282
  gc.collect()
283
+ return sd_image_path, seed
284
 
285
  @spaces.GPU(duration=60)
286
  def generate_60(
 
334
  sd_image_path = f"rv50_B_{seed}.png"
335
  rv_image.save(sd_image_path,optimize=False,compress_level=0)
336
  upload_to_ftp(sd_image_path)
337
+ #image_paths = save_image(rv_image)
338
  torch.cuda.empty_cache()
339
  gc.collect()
340
+ return sd_image_path, seed
341
 
342
  @spaces.GPU(duration=90)
343
  def generate_90(
 
391
  sd_image_path = f"rv50_B_{seed}.png"
392
  rv_image.save(sd_image_path,optimize=False,compress_level=0)
393
  upload_to_ftp(sd_image_path)
394
+ #image_paths = save_image(rv_image)
395
  torch.cuda.empty_cache()
396
  gc.collect()
397
+ return sd_image_path, seed
398
 
399
  def load_predefined_images1():
400
  predefined_images1 = [