Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ def predict_step(image_paths):
|
|
29 |
preds = [pred.strip() for pred in preds]
|
30 |
return preds
|
31 |
|
32 |
-
torch.hub.download_url_to_file('https://github.com/AaronCWacker/Yggdrasil/blob/main/images/35-Favorite-Games.jpg', '35-Favorite-Games.jpg')
|
33 |
|
34 |
#result = predict_step(['35-Favorite-Games.jpg'])
|
35 |
|
@@ -47,7 +47,8 @@ article = "nlpconnect/vit-gpt2-image-captioning"
|
|
47 |
|
48 |
input = gr.inputs.Image(label="Upload your Image", type = 'pil', optional=True)
|
49 |
output = gr.outputs.Textbox(type="auto",label="Captions")
|
50 |
-
examples = [['35-Favorite-Games.jpg']]
|
|
|
51 |
|
52 |
interface = gr.Interface(
|
53 |
fn=predict,
|
|
|
29 |
preds = [pred.strip() for pred in preds]
|
30 |
return preds
|
31 |
|
32 |
+
#torch.hub.download_url_to_file('https://github.com/AaronCWacker/Yggdrasil/blob/main/images/35-Favorite-Games.jpg', '35-Favorite-Games.jpg')
|
33 |
|
34 |
#result = predict_step(['35-Favorite-Games.jpg'])
|
35 |
|
|
|
47 |
|
48 |
input = gr.inputs.Image(label="Upload your Image", type = 'pil', optional=True)
|
49 |
output = gr.outputs.Textbox(type="auto",label="Captions")
|
50 |
+
#examples = [['35-Favorite-Games.jpg']]
|
51 |
+
examples = [f"{i}.jpg" for i in range(1,7)]
|
52 |
|
53 |
interface = gr.Interface(
|
54 |
fn=predict,
|