Spaces:

diamantrsd
/

image-captioning-bahasa-indonesia

Runtime error

diamantrsd commited on Sep 27, 2023

Commit

1355606

•

1 Parent(s): e2452eb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 def generate_caption(input_image):
     # Mengonversi gambar ke mode RGB
     image = input_image.convert('RGB')
-    inputs = processor(images=image, return_tensors="pt")
     # inisial variabel inputs
     inputs["max_length"] = 20
     inputs["num_beams"] = 1

 def generate_caption(input_image):
     # Mengonversi gambar ke mode RGB
     image = input_image.convert('RGB')
+    inputs = caption_processor(images=image, return_tensors="pt")
     # inisial variabel inputs
     inputs["max_length"] = 20
     inputs["num_beams"] = 1