alex6095 commited on
Commit
38eb205
·
1 Parent(s): 24f3a33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,6 +108,6 @@ if text:
108
  _, indices = torch.topk(pred, 3)
109
  pred_print = []
110
  for i in indices.squeeze(0):
111
- year, month = i2ym(i.item()))
112
  pred_print.append(year+"-"+month)
113
  st.write(", ".join(pred_print))
 
108
  _, indices = torch.topk(pred, 3)
109
  pred_print = []
110
  for i in indices.squeeze(0):
111
+ year, month = i2ym(i.item())
112
  pred_print.append(year+"-"+month)
113
  st.write(", ".join(pred_print))