Spaces:

yongyeol
/

imagetoaudio

Runtime error

yongyeol commited on Jul 7

Commit

7d967cc

verified ·

1 Parent(s): 725cad1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,6 +3,7 @@ from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoToken
 from PIL import Image
 import torch
 import requests
 # Load caption model
 caption_model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")

 from PIL import Image
 import torch
 import requests
+import os
 # Load caption model
 caption_model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")