refactor: adjust python client example

python_client.py

@@ -3,10 +3,21 @@ import base64
 from PIL import Image
 from io import BytesIO
 import matplotlib.pyplot as plt
+import argparse
+parser = argparse.ArgumentParser(description="Client for FastAPI image inference")
+parser.add_argument("--env", '-e', type=str, default="local", help="Environment: local or deployed")
+args = parser.parse_args()
+
+if args.env == "local":
+    FASTAPI_URL = "http://localhost:8000"
+elif args.env == "deployed":
+    FASTAPI_URL = "https://riciii7-tumor-detection-fastapi.hf.space"
+else: 
+    raise ValueError("Invalid environment. Choose 'local' or 'deployed'.")
 
 with open('glioma.jpg', 'rb') as f:
     files = {'file': ('glioma.jpg', f, 'image/jpeg')}
-    response = requests.post('http://localhost:8000/inference', files=files)
+    response = requests.post(f'{FASTAPI_URL}/inference', files=files)
 
 result = response.json()