Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,4 +6,10 @@ from PIL import Image
|
|
| 6 |
processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
| 7 |
model = BlipFConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
|
| 8 |
|
| 9 |
-
our_image = Image.open("")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
| 7 |
model = BlipFConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
|
| 8 |
|
| 9 |
+
our_image = Image.open("elephant.jpg")
|
| 10 |
+
inputs = processor(our_image, return_tensors = "pt")
|
| 11 |
+
outputs = model.generate(**inputs)
|
| 12 |
+
|
| 13 |
+
caption = processor.decode(outputs[0], skip_special_tokens = True)
|
| 14 |
+
print(caption)
|
| 15 |
+
|