Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
"""
|
| 2 |
-
|
| 3 |
Fixed for ZeroGPU with lazy loading
|
| 4 |
"""
|
| 5 |
|
|
@@ -82,7 +82,7 @@ def create_blend_weights(tile_size, overlap):
|
|
| 82 |
|
| 83 |
|
| 84 |
def build_model(device):
|
| 85 |
-
"""Build and load the
|
| 86 |
torch = get_torch()
|
| 87 |
UNet, GaussianDiffusion = get_model_modules()
|
| 88 |
from huggingface_hub import hf_hub_download
|
|
@@ -367,9 +367,9 @@ with gr.Blocks(title="Axion - SAR to Optical") as demo:
|
|
| 367 |
gr.HTML("""
|
| 368 |
<div style="font-size: 0.8rem; color: #666; padding: 8px 12px; background: rgba(255,255,255,0.03); border-radius: 6px; margin: 8px 0;">
|
| 369 |
<strong style="color: #888;">Input Guidelines:</strong><br>
|
| 370 |
-
|
| 371 |
-
|
| 372 |
-
|
| 373 |
</div>
|
| 374 |
""")
|
| 375 |
with gr.Row():
|
|
|
|
| 1 |
"""
|
| 2 |
+
Axion: SAR-to-Optical Translation - HuggingFace Space
|
| 3 |
Fixed for ZeroGPU with lazy loading
|
| 4 |
"""
|
| 5 |
|
|
|
|
| 82 |
|
| 83 |
|
| 84 |
def build_model(device):
|
| 85 |
+
"""Build and load the Axion model."""
|
| 86 |
torch = get_torch()
|
| 87 |
UNet, GaussianDiffusion = get_model_modules()
|
| 88 |
from huggingface_hub import hf_hub_download
|
|
|
|
| 367 |
gr.HTML("""
|
| 368 |
<div style="font-size: 0.8rem; color: #666; padding: 8px 12px; background: rgba(255,255,255,0.03); border-radius: 6px; margin: 8px 0;">
|
| 369 |
<strong style="color: #888;">Input Guidelines:</strong><br>
|
| 370 |
+
• Use raw SAR imagery (single-band grayscale)<br>
|
| 371 |
+
• VV polarization preferred, VH also supported<br>
|
| 372 |
+
• Any resolution supported (processed in 256×256 tiles)
|
| 373 |
</div>
|
| 374 |
""")
|
| 375 |
with gr.Row():
|