Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -187,7 +187,6 @@ model = torch.load('./BestModel_20000_Datos', map_location=device)
|
|
187 |
start_token = word_to_index['<start>']
|
188 |
end_token = word_to_index['<end>']
|
189 |
pad_token = word_to_index['<pad>']
|
190 |
-
max_seq_len = 67
|
191 |
|
192 |
def predict(inp):
|
193 |
device = "cpu"
|
@@ -207,4 +206,6 @@ def predict(inp):
|
|
207 |
|
208 |
gr.Interface(fn=predict,
|
209 |
inputs=gr.Image(type="filepath"),
|
210 |
-
outputs=gr.Text()
|
|
|
|
|
|
187 |
start_token = word_to_index['<start>']
|
188 |
end_token = word_to_index['<end>']
|
189 |
pad_token = word_to_index['<pad>']
|
|
|
190 |
|
191 |
def predict(inp):
|
192 |
device = "cpu"
|
|
|
206 |
|
207 |
gr.Interface(fn=predict,
|
208 |
inputs=gr.Image(type="filepath"),
|
209 |
+
outputs=gr.Text(),
|
210 |
+
title = "Clothe captioning model",
|
211 |
+
description = "A clothe image captioning model to get descriptions of your code.\n Take your phone, make a picture of your clothes, upload it and you are ready to go").launch()
|