Update app.py

app.py

@@ -227,13 +227,9 @@ def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
         f.write(f"Steps: {num_inference_steps} \n")
         f.write(f"Guidance Scale: {guidance_scale} \n")
         f.write(f"SPACE SETUP: \n")
-        f.write(f"Use Model Dtype: no \n")
-        f.write(f"Model Scheduler: Euler_a all_custom before cuda \n")
-        f.write(f"Model VAE: sdxl-vae to bfloat safetensor=false before cuda then attn_proc / scale factor 8 \n")
         f.write(f"Model UNET: ford442/RealVisXL_V5.0_BF16 \n")
     upload_to_ftp(filename) 
 
-
 def captioning(img):
     prompts_array = [
         "Adjectives describing this scene are:",
@@ -246,15 +242,12 @@ def captioning(img):
         "The setting of this scene must be located", 
     # Add more prompts here
     ]
-
     output_prompt=[]
-
     # Initial caption generation without a prompt:
     inputsa = processor5(images=img, return_tensors="pt").to('cuda')
     generated_ids = model5.generate(**inputsa, min_length=42, max_length=42)
     generated_text = processor5.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
     print(generated_text)  
-
     # Loop through prompts array:
     for prompt in prompts_array:
         inputs = processor5(images=img, text=prompt, return_tensors="pt").to('cuda')
@@ -263,7 +256,6 @@ def captioning(img):
         response_text = generated_text.replace(prompt, "").strip() #Or could try .split(prompt, 1)[-1].strip()
         output_prompt.append(response_text)
         print(f"{response_text}\n") # Print only the response text
-
     # Continue conversation:
     inputf = processor5(images=img, text=generated_text + 'So therefore', return_tensors="pt").to('cuda')
     generated_ids = model5.generate(**inputf, max_length=42)
@@ -274,7 +266,6 @@ def captioning(img):
     print(output_prompt)
     return output_prompt
 
-    
 def expand_prompt(prompt):
         system_prompt_rewrite = (
             "You are an AI assistant that rewrites image prompts to be more descriptive and detailed."
@@ -326,7 +317,7 @@ def expand_prompt(prompt):
         print('-- filtered prompt 2 --')
         print(enhanced_prompt_2)
         enh_prompt=[enhanced_prompt,enhanced_prompt_2]
-    return enh_prompt 
+        return enh_prompt 
     
 @spaces.GPU(duration=40)
 def generate_30(