Commit
·
5e2bbd7
1
Parent(s):
697f4c6
Update app.py
Browse files
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 |
-
|
60 |
|
61 |
if torch.cuda.is_available():
|
62 |
torch.cuda.empty_cache()
|
63 |
|
64 |
-
return
|
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:
|