Update app.py

app.py

@@ -15,11 +15,11 @@ def predict(text):
     encoded_text = encoder(text)
     #print(encoded_text)
     prediction = (model.predict(encoded_text))
-    print(prediction)
-    prediction = np.round(prediction)
-    if prediction==1:
-        return "Disaster"
-    return "Not a Disaster"
+    return prediction
+    #prediction = np.round(prediction)
+    #if prediction==1:
+    #    return "Disaster"
+    #return "Not a Disaster"
 
 title="Relevance Classifier"
 description="<p style='text-align:center'>Classifies input text into Disaster-related or not disaster related."