manual repo input

app.py

@@ -34,8 +34,8 @@ def get_static_map_image(lat_lon, api):
 
 
 @st.cache_resource(show_spinner=False)
-def load_model():
-    model_path = hf_hub_download(repo_id="Vannsh/v8x-obb", filename="obb3.pt")
+def load_model(repo_id, filename):
+    model_path = hf_hub_download(repo_id=repo_id, filename=filename)
     model = YOLO(model_path, task="obb")
     return model
 
@@ -56,6 +56,11 @@ else:
         state.GMS_KEY = api_key
         st.rerun()
 
+# get two columns
+col1, col2 = st.columns(2)
+repo = col1.text_input("HF Repo", "Vannsh/v8x-obb")
+path = col2.text_input("Model Path", "obb3.pt")
+
 #### Get Latitude, Longitude
 lat_lon = st.text_input("Enter Latitude, Longitude (comma separated)", placeholder="28.00,77.00")
 if lat_lon == "":
@@ -86,7 +91,7 @@ buffer.seek(0)
 st.image(image, caption=lat_lon, use_column_width=True)
 
 ### Predict
-model = load_model()
+model = load_model(repo, path)
 result = model(image)[0]
 detections = sv.Detections.from_ultralytics(result)