rifatramadhani commited on
Commit
5f2cf96
·
verified ·
1 Parent(s): 3844f0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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