Update app.py

app.py

@@ -1,17 +1,25 @@
 import joblib
 import gradio as gr
+import numpy as np
 
 # Load model
 model = joblib.load("./scaler.pkl")
 
 # Prediction function
 def predict(val1, val2):
-    prediction = model.predict([[float(val1), float(val2)]])
-    return str(prediction[0])
+    try:
+        data = np.array([[float(val1), float(val2)]])
+        prediction = model.predict(data)
+        return str(prediction[0])
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 # UI
 gr.Interface(
     fn=predict,
-    inputs=["number", "number"],
+    inputs=[
+        gr.Number(label="CGPA"),
+        gr.Number(label="IQ")
+    ],
     outputs="text"
-).launch()
+).launch()