Juliofc commited on
Commit
b9b13c1
·
1 Parent(s): 7bdd103

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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()).launch(debug=True)
 
 
 
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()