Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ def classify(query):
|
|
29 |
|
30 |
start_time = datetime.datetime.now()
|
31 |
|
32 |
-
result = sentiment_task(data, batch_size=128)
|
33 |
|
34 |
end_time = datetime.datetime.now()
|
35 |
elapsed_time = end_time - start_time
|
|
|
29 |
|
30 |
start_time = datetime.datetime.now()
|
31 |
|
32 |
+
result = sentiment_task(data, batch_size=128, top_k=3)
|
33 |
|
34 |
end_time = datetime.datetime.now()
|
35 |
elapsed_time = end_time - start_time
|