azamat commited on
Commit
f19a791
·
1 Parent(s): e7f40dd
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -35,7 +35,8 @@ def predict_coordinates(text):
35
  encoding = tokenizer(text, padding="max_length", truncation=True, \
36
  max_length=128, return_tensors='pt')
37
  outputs = coordinates_model(**encoding)
38
- return outputs[0][0], outputs[0][1]
 
39
 
40
  def predict(text):
41
  text = process_tweet(text)
 
35
  encoding = tokenizer(text, padding="max_length", truncation=True, \
36
  max_length=128, return_tensors='pt')
37
  outputs = coordinates_model(**encoding)
38
+ print(outputs)
39
+ return outputs[0][0][0], outputs[0][0][1]
40
 
41
  def predict(text):
42
  text = process_tweet(text)