ford442 commited on
Commit
6cf9565
·
verified ·
1 Parent(s): 0eece6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -274,12 +274,14 @@ def generate_30(
274
  uploadNote(prompt,num_inference_steps,guidance_scale,timestamp,denoise)
275
  batch_options = options.copy()
276
  rv_image = pipe(**batch_options).images[0]
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)
@@ -331,12 +333,14 @@ def generate_60(
331
  uploadNote(prompt,num_inference_steps,guidance_scale,timestamp,denoise)
332
  batch_options = options.copy()
333
  rv_image = pipe(**batch_options).images[0]
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)
@@ -388,12 +392,14 @@ def generate_90(
388
  uploadNote(prompt,num_inference_steps,guidance_scale,timestamp,denoise)
389
  batch_options = options.copy()
390
  rv_image = pipe(**batch_options).images[0]
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():
 
274
  uploadNote(prompt,num_inference_steps,guidance_scale,timestamp,denoise)
275
  batch_options = options.copy()
276
  rv_image = pipe(**batch_options).images[0]
277
+ sd_image_path = f"rv50_B_{timestamp}.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
+ unique_name = str(uuid.uuid4()) + ".png"
284
+ os.symlink(sd_image_path, unique_name)
285
  return sd_image_path, seed
286
 
287
  @spaces.GPU(duration=60)
 
333
  uploadNote(prompt,num_inference_steps,guidance_scale,timestamp,denoise)
334
  batch_options = options.copy()
335
  rv_image = pipe(**batch_options).images[0]
336
+ sd_image_path = f"rv50_B_{timestamp}.png"
337
  rv_image.save(sd_image_path,optimize=False,compress_level=0)
338
  upload_to_ftp(sd_image_path)
339
  #image_paths = save_image(rv_image)
340
  #torch.cuda.empty_cache()
341
  #gc.collect()
342
+ unique_name = str(uuid.uuid4()) + ".png"
343
+ os.symlink(sd_image_path, unique_name)
344
  return sd_image_path, seed
345
 
346
  @spaces.GPU(duration=90)
 
392
  uploadNote(prompt,num_inference_steps,guidance_scale,timestamp,denoise)
393
  batch_options = options.copy()
394
  rv_image = pipe(**batch_options).images[0]
395
+ sd_image_path = f"rv50_B_{timestamp}.png"
396
  rv_image.save(sd_image_path,optimize=False,compress_level=0)
397
  upload_to_ftp(sd_image_path)
398
  #image_paths = save_image(rv_image)
399
  #torch.cuda.empty_cache()
400
  #gc.collect()
401
+ unique_name = str(uuid.uuid4()) + ".png"
402
+ os.symlink(sd_image_path, unique_name)
403
  return sd_image_path, seed
404
 
405
  def load_predefined_images1():