Update app.py

app.py

@@ -50,6 +50,7 @@ def predict_model(qost, qgst):
     plt.xlabel("QOST") ;plt.ylabel("QGST")
     plt.legend(True,handles=legend_items)
     plt.title("Original Dataset"); plt.title("Predicted Dataset")
+    plt.ylim(0,200)
     return plt, regime[ int(_pred[0])-1  ]
 interface = gr.Interface(predict_model, inputs=[gr.Number(label="QOil[bbl/day]", value=20000), gr.Number(label="QGas[mmscf/day]", value=100)], 
                          outputs=[gr.Plot(label="Flow Map"),gr.Textbox(label="Identified Regime")])