awacke1 commited on
Commit
56286b1
·
1 Parent(s): 8c56a9e

Update backup.app.py

Browse files
Files changed (1) hide show
  1. backup.app.py +12 -4
backup.app.py CHANGED
@@ -27,11 +27,19 @@ def set_model(current_model):
27
  current_model = models[current_model]
28
  return gr.update(label=(f"{current_model}"))
29
 
30
-
31
  def send_it1(inputs, model_choice):
32
- proc1=models2[model_choice]
33
- output1=proc1(inputs)
34
- return(output1)
 
 
 
 
 
 
 
 
 
35
  css=""""""
36
 
37
 
 
27
  current_model = models[current_model]
28
  return gr.update(label=(f"{current_model}"))
29
 
 
30
  def send_it1(inputs, model_choice):
31
+ proc1 = models2[model_choice]
32
+ output1 = proc1(inputs)
33
+ if not os.path.exists('saved_prompts'):
34
+ os.makedirs('saved_prompts')
35
+ if not os.path.exists('saved_images'):
36
+ os.makedirs('saved_images')
37
+ safe_filename = slugify(inputs)[:50] # Truncate to first 50 characters
38
+ with open(f'saved_prompts/{safe_filename}.txt', 'w') as f:
39
+ f.write(inputs)
40
+ output1.save(f'saved_images/{safe_filename}.png')
41
+ return output1
42
+
43
  css=""""""
44
 
45