Update app.py

app.py

@@ -275,9 +275,11 @@ def inference():
         dl_prob = np.expand_dims(dl_prob, axis=0)
         lc_prob = LC_model.predict_proba(lc_data)
         feature = DR_model.transform(np.concatenate([dl_prob, lc_prob], axis=1))
-        final_p = TF_model.predict_proba(feature)
-        final_p = np.round(final_p[0], decimals=2)
-        return {'急性期': final_p[0], '亚急性期': final_p[1], '慢性期': final_p[2]}
+        final_p = TF_model.predict_proba(feature)[0]
+        p0 = np.round(float(final_p[0]), decimals=2)
+        p1 = np.round(float(final_p[1]), decimals=2)
+        p2 = np.round(float(final_p[2]), decimals=2)
+        return {'慢性期': p2,'亚急性期': p1,'急性期': p0 }
     except:
         return {'急性期': 0.15, '亚急性期': 0.2, '慢性期': 0.3}