azamat commited on
Commit
e0e99ed
1 Parent(s): d9b2223
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ 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 round(outputs[0][0][0], 3), round(outputs[0][0][1], 3)
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
+ return round(outputs[0][0][0].item(), 3), round(outputs[0][0][1].item(), 3)
39
 
40
  def predict(text):
41
  text = process_tweet(text)