andreinigo commited on
Commit
5e2bbd7
·
1 Parent(s): 697f4c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -56,12 +56,12 @@ def generate_caption(image):
56
  draw.text((x, y), line, fill=(255, 255, 255), font=font)
57
  y += line_height
58
 
59
- meme_image = meme_image.convert('RGB')
60
 
61
  if torch.cuda.is_available():
62
  torch.cuda.empty_cache()
63
 
64
- return meme_image
65
 
66
 
67
  with gr.Blocks() as demo:
 
56
  draw.text((x, y), line, fill=(255, 255, 255), font=font)
57
  y += line_height
58
 
59
+ pil_image = pil_image.convert('RGB')
60
 
61
  if torch.cuda.is_available():
62
  torch.cuda.empty_cache()
63
 
64
+ return pil_image
65
 
66
 
67
  with gr.Blocks() as demo: