Iris prediction

app.py

@@ -1,18 +1,30 @@
 import gradio as gr
+import hopsworks
+import joblib
+import pandas as pd
+import numpy as np
+project = hopsworks.login()
+fs = project.get_feature_store()
 
-def predict_length(sep_length, sep_width, pet_length, pet_width):
-    return f"Sepal Length: {sep_length}, Sepal Width: {sep_width}, Petal Length: {pet_length}, Petal Width: {pet_width}"    
+mr = project.get_model_registry()
+model = mr.get_model("iris_model", version=1)
+model_dir = model.download()
+model = joblib.load(model_dir + "/iris_model.pkl")
+feature_view = fs.get_feature_view(name="iris", version=1)
+batch_data = feature_view.get_batch_data()
 
-iface = gr.Interface(
-    fn=predict_length, 
+def greet(sep_length, sep_width, pet_length, pet_width):
+    df = pd.DataFrame({ "sepal_length": sep_length, "sepal_width": sep_width, "petal_length": pet_length, "petal_width": pet_width}, index=[0])
+    prediction = model.predict(df)[0]
+    return prediction, f"Images\{prediction}.jpg"
+
+predict = gr.Interface(
+    fn=greet,
     inputs=[
-        gr.Textbox(label="Sepal Length"),
-        gr.Textbox(label="Sepal Width"),
-        gr.Textbox(label="Petal Length"),
-        gr.Textbox(label="Petal Width"),
-    ], 
-    outputs="text",
-    live=True
+        gr.Textbox(placeholder="Sepal length here", label = "Sepal Length"),
+        gr.Textbox(placeholder="Sepal width here", label = "Sepal Width"),
+        gr.Textbox(placeholder="Petal length here", label = "Petal Length"),
+        gr.Textbox(placeholder="Petal width here", label = "Petal Width")],
+    outputs=[gr.Textbox(label = "Prediction"), gr.Image(type="pil", label="Iris")],
 )
-
-iface.launch()
+predict.launch()

flagged/log.csv

@@ -0,0 +1,4 @@
+Sepal Length,Sepal Width,Petal Length,Petal Width,output,flag,username,timestamp
+1,2,nu3,number,"Sepal Length: 1, Sepal Width: 2, Petal Length: nu3, Petal Width: number",,,2023-11-12 13:07:01.911729
+1,2,nu3,number,"Sepal Length: 1, Sepal Width: 2, Petal Length: nu3, Petal Width: number",,,2023-11-12 13:07:02.351389
+1,2,nu3,number,"Sepal Length: 1, Sepal Width: 2, Petal Length: nu3, Petal Width: number",,,2023-11-12 13:07:02.524937