tst inference

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import joblib
 import requests
 import os
+import pandas as pd
+
 
 # Download the model file
 model_url = "https://huggingface.co/stummala521/AI4SAR/resolve/main/model.pkl"
@@ -15,12 +17,9 @@ with open(local_filename, 'wb') as f:
 # Load the model
 model = joblib.load(local_filename)
 
-test_data = [{
-    "location_found_elevation": 1500,
-    "situation": "lost"
-}]
+test_data = pd.DataFrame({'location_found_elevation': [0, 0, 0], 'situation': ['fake', 'fake', 'fake']})
 
-output = model.predict([test_data])
+output = model.predict(test_data)
 
 def greet(name):
     return "Hello " + name + "!" + str(output[0])

requirements.txt

@@ -1,2 +1,3 @@
 joblib==1.4.2
-skops==0.10.0
+skops==0.10.0
+pandas==2.2.1