tst inference

app.py

@@ -1,11 +1,19 @@
 import gradio as gr
 import joblib
-from skops import hub_utils
+import requests
+import os
 
-hub_utils.download(repo_id="stummala521/AI4SAR", dst="model.pkl")
-model = joblib.load(
-	"model.pkl"
-)
+# Download the model file
+model_url = "https://huggingface.co/stummala521/AI4SAR/resolve/main/model.pkl"
+local_filename = "model.pkl"
+
+# Download the file
+response = requests.get(model_url, allow_redirects=True)
+with open(local_filename, 'wb') as f:
+    f.write(response.content)
+
+# Load the model
+model = joblib.load(local_filename)
 
 test_data = {
     "location_found_elevation": 1500,
@@ -15,8 +23,7 @@ test_data = {
 output = model.predict([test_data])
 
 def greet(name):
-    return "Hello " + name + "!" + output
+    return "Hello " + name + "!" + str(output[0])
 
 demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
-
+demo.launch()