Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ def predict_sentiment(sentence = "Volim ti"):
|
|
45 |
features_dict = convert_to_features(dataset_dict, convert_func_dict)
|
46 |
|
47 |
predictions = []
|
48 |
-
|
49 |
for _, batch in enumerate(features_dict["document"]['test']):
|
50 |
for key, value in batch.items():
|
51 |
batch[key] = batch[key].to(device)
|
|
|
45 |
features_dict = convert_to_features(dataset_dict, convert_func_dict)
|
46 |
|
47 |
predictions = []
|
48 |
+
|
49 |
for _, batch in enumerate(features_dict["document"]['test']):
|
50 |
for key, value in batch.items():
|
51 |
batch[key] = batch[key].to(device)
|