Upload app.py with huggingface_hub

app.py

@@ -581,10 +581,13 @@ def load_sar_image(filepath):
     return Image.open(filepath).convert('RGB')
 
 
-def translate_sar(image, num_steps, overlap, enhance):
+def translate_sar(file, num_steps, overlap, enhance):
     """Main translation function."""
     global model
     
+    if file is None:
+        return None, None, "Please upload a SAR image"
+    
     if model is None:
         print("Loading model...")
         model = E3DiffHighRes()
@@ -592,9 +595,9 @@ def translate_sar(image, num_steps, overlap, enhance):
     
     print("Processing image...")
     
-    # Handle file upload
-    if isinstance(image, str):
-        image = load_sar_image(image)
+    # Handle file upload - get the filepath
+    filepath = file.name if hasattr(file, 'name') else file
+    image = load_sar_image(filepath)
     
     w, h = image.size
     print(f"Input size: {w}x{h}")
@@ -644,7 +647,7 @@ with gr.Blocks(title="E3Diff: SAR-to-Optical Translation") as demo:
     
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(label="SAR Input", type="pil")
+            input_file = gr.File(label="SAR Input (TIFF, PNG, JPG supported)", file_types=[".tif", ".tiff", ".png", ".jpg", ".jpeg"])
             
             with gr.Row():
                 num_steps = gr.Slider(1, 8, value=1, step=1, label="Quality Steps (1=fast, 4-8=high quality)")
@@ -661,7 +664,7 @@ with gr.Blocks(title="E3Diff: SAR-to-Optical Translation") as demo:
     
     submit_btn.click(
         fn=translate_sar,
-        inputs=[input_image, num_steps, overlap, enhance],
+        inputs=[input_file, num_steps, overlap, enhance],
         outputs=[output_image, output_file, info_text]
     )