Update main.py

main.py

@@ -8,12 +8,12 @@ app = FastAPI()
 
 # Directory where images will be saved
 # This directory must be publicly accessible through your web server
-image_directory = "/images"
-os.makedirs(image_directory, exist_ok=True)
+#image_directory = "/images"
+#os.makedirs(image_directory, exist_ok=True)
 
 # Base URL for the images
 # Replace with your server's base URL
-base_url = "https://zliang-argoapi.hf.space/images"
+base_url = "https://zliang-argoapi.hf.space"
 
 @app.get("/plot_trajectory")
 def plot_trajectory(lon_min: float, lon_max: float, lat_min: float, lat_max: float, 
@@ -29,8 +29,7 @@ def plot_trajectory(lon_min: float, lon_max: float, lat_min: float, lat_max: flo
 
         # Save the plot to a file with a unique name
         filename = str(uuid.uuid4()) + ".png"
-        file_path = os.path.join(image_directory, filename)
-        plt.savefig(file_path)
+        plt.savefig(filename)
         plt.close()
 
         # Construct the URL for the saved image