1inkusFace commited on
Commit
ef032f6
·
verified ·
1 Parent(s): 8b2cdfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -430,7 +430,7 @@ def generate_30(
430
  del processor5
431
  gc.collect()
432
  torch.cuda.empty_cache()
433
- expanded = expand_prompt(caption)
434
  new_prompt = prompt+' '+expanded
435
  print("-- ------------ --")
436
  print("-- FINAL PROMPT --")
@@ -580,7 +580,7 @@ def generate_60(
580
  del processor5
581
  gc.collect()
582
  torch.cuda.empty_cache()
583
- expanded = expand_prompt(caption)
584
  new_prompt = prompt+' '+expanded
585
  print("-- ------------ --")
586
  print("-- FINAL PROMPT --")
@@ -733,7 +733,7 @@ def generate_90(
733
  del processor5
734
  gc.collect()
735
  torch.cuda.empty_cache()
736
- expanded = expand_prompt(caption)
737
  new_prompt = prompt+' '+expanded
738
  print("-- ------------ --")
739
  print("-- FINAL PROMPT --")
 
430
  del processor5
431
  gc.collect()
432
  torch.cuda.empty_cache()
433
+ expanded = expand_prompt(captions)
434
  new_prompt = prompt+' '+expanded
435
  print("-- ------------ --")
436
  print("-- FINAL PROMPT --")
 
580
  del processor5
581
  gc.collect()
582
  torch.cuda.empty_cache()
583
+ expanded = expand_prompt(captions)
584
  new_prompt = prompt+' '+expanded
585
  print("-- ------------ --")
586
  print("-- FINAL PROMPT --")
 
733
  del processor5
734
  gc.collect()
735
  torch.cuda.empty_cache()
736
+ expanded = expand_prompt(captions)
737
  new_prompt = prompt+' '+expanded
738
  print("-- ------------ --")
739
  print("-- FINAL PROMPT --")