ki33elev commited on
Commit
8393245
1 Parent(s): af657dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def predict(title, summary, tokenizer, model):
39
  @st.cache(suppress_st_warning=True)
40
  def get_results(prediction, prediction_probs):
41
  frame = pd.DataFrame({'Topic': prediction, 'Confidence': prediction_probs})
42
- frame.index = np.arange(1, len(frame))
43
  return frame
44
 
45
  label_to_theme = {0: 'Computer science', 1: 'Economics', 2: 'Electrical Engineering and Systems Science', 3: 'Math',
 
39
  @st.cache(suppress_st_warning=True)
40
  def get_results(prediction, prediction_probs):
41
  frame = pd.DataFrame({'Topic': prediction, 'Confidence': prediction_probs})
42
+ frame.index = np.arange(1, len(frame) + 1)
43
  return frame
44
 
45
  label_to_theme = {0: 'Computer science', 1: 'Economics', 2: 'Electrical Engineering and Systems Science', 3: 'Math',