apratim24 commited on
Commit
b32b280
·
verified ·
1 Parent(s): 1a3a8f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint)
14
  def generate_story(image, genre, style):
15
  try:
16
  # Preprocess the image
17
- image = Image.open(image.name).convert('RGB')
18
  image_features = feature_extractor(images=image, return_tensors="pt")
19
 
20
  # Generate image caption
 
14
  def generate_story(image, genre, style):
15
  try:
16
  # Preprocess the image
17
+ image = image.convert('RGB')
18
  image_features = feature_extractor(images=image, return_tensors="pt")
19
 
20
  # Generate image caption