Update app.py

app.py

@@ -13,8 +13,8 @@ from functions import *
 
 project = hopsworks.login()
 	
-
-def air_quality(city):
+models=["Gradient_Boost","Random_Forest","Lasso"]## Add others models names as in hopsworks
+def air_quality(city,model_name):
     start_date = datetime.now() - timedelta(days=1)
     start_time = int(start_date.timestamp()) * 1000
     X = pd.read_csv('x.csv')
@@ -24,7 +24,7 @@ def air_quality(city):
     X = X.drop(X.columns[0],axis=1)
 
     mr = project.get_model_registry()
-    model = mr.get_model("gradient_boost_paris_model", version=1)
+    model = mr.get_model(model_name, version=1)
     model_dir = model.download()
     model = joblib.load(model_dir + "/model.pkl")
 	    
@@ -39,14 +39,20 @@ def air_quality(city):
     return predictions
 
 
-
-	
+with gr.Blocks() as demo:
+    gr.Label("Predict air quality in Paris")
+    with gr.Row():
+        inp = ['text',
+               gr.Radio(models, value="Gradient_Boost")]
+        out = gr.DataFrame()
+    btn = gr.Button("Run prediction")
+    btn.click(fn=air_quality, inputs=inp, outputs='text')
 	
     
-demo = gr.Interface(fn=air_quality, title="Air quality predictor",
-description="Input a value to get next weeks AQI prediction for Paris", inputs="text", outputs="text")
+#demo = gr.Interface(fn=air_quality, title="Air quality predictor",
+#description="Input a value to get next weeks AQI prediction for Paris", inputs="text", outputs="text")
 	
 
 	    
-if __name__ == "__main__":
-	demo.launch()
+if _name_ == "_main_":
+	demo.launch()