retromarz commited on
Commit
f40c282
·
verified ·
1 Parent(s): f907e73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)) # Small resolution
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."