Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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))
|