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