Update app.py

app.py

@@ -144,15 +144,15 @@ def fragment_single_abstract(abstract):
     predictions = tf.argmax(model.predict(dataset), axis=1)
 
     for i, prediction in enumerate(predictions):
-        if prediction == 0:
+        if prediction == 3:
             objective.append(sentences_original[i])
         elif prediction == 1:
             methods.append(sentences_original[i])
-        elif prediction == 2:
+        elif prediction == 4:
             results.append(sentences_original[i])
-        elif prediction == 3:
+        elif prediction == 0:
             conclusion.append(sentences_original[i])
-        elif prediction == 4:
+        elif prediction == 2:
             background.append(sentences_original[i])
 
     end_time = timer()