mdnestor commited on
Commit
0a589ca
1 Parent(s): 1616986

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -245,6 +245,7 @@ class InferenceModel(object):
245
  tokens = tokens[:np.argmax(tokens == vocabularies.DECODED_EOS_ID)]
246
  return tokens
247
 
 
248
  inference_model = InferenceModel('/home/user/app/checkpoints/mt3/', 'mt3')
249
 
250
  def inference(url):
 
245
  tokens = tokens[:np.argmax(tokens == vocabularies.DECODED_EOS_ID)]
246
  return tokens
247
 
248
+ print(glob.glob("."))
249
  inference_model = InferenceModel('/home/user/app/checkpoints/mt3/', 'mt3')
250
 
251
  def inference(url):