Update app.py
Browse files
app.py
CHANGED
@@ -70,7 +70,7 @@ def generate_caption(input_image: Image.Image, caption_type: str = "descriptive"
|
|
70 |
|
71 |
try:
|
72 |
# Resize image to reduce memory usage
|
73 |
-
input_image = input_image.resize((256, 256))
|
74 |
|
75 |
# Prepare the prompt
|
76 |
prompt = f"Write a {caption_length} {caption_type} caption for this image."
|
|
|
70 |
|
71 |
try:
|
72 |
# Resize image to reduce memory usage
|
73 |
+
input_image = input_image.resize((256, 256))
|
74 |
|
75 |
# Prepare the prompt
|
76 |
prompt = f"Write a {caption_length} {caption_type} caption for this image."
|