1inkusFace commited on
Commit
8f45cc1
Β·
verified Β·
1 Parent(s): 350939b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -236,6 +236,7 @@ def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
236
  f.write(f"Model UNET: ford442/RealVisXL_V5.0_BF16 \n")
237
  upload_to_ftp(filename)
238
 
 
239
  def captioning(img):
240
  prompts_array = [
241
  "Adjectives describing this scene are:",
 
236
  f.write(f"Model UNET: ford442/RealVisXL_V5.0_BF16 \n")
237
  upload_to_ftp(filename)
238
 
239
+ @torch.no_grad()
240
  def captioning(img):
241
  prompts_array = [
242
  "Adjectives describing this scene are:",