Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,7 +64,7 @@ aura_sr_v1 = ZeroGPUAuraSR.from_pretrained("fal-ai/AuraSR")
|
|
| 64 |
|
| 65 |
@spaces.GPU()
|
| 66 |
def predict(img, model_selection):
|
| 67 |
-
return {'v1': aura_sr_v1, 'v2': aura_sr}.get().upscale_4x(img)
|
| 68 |
|
| 69 |
|
| 70 |
demo = gr.Interface(
|
|
|
|
| 64 |
|
| 65 |
@spaces.GPU()
|
| 66 |
def predict(img, model_selection):
|
| 67 |
+
return {'v1': aura_sr_v1, 'v2': aura_sr}.get(model_selection).upscale_4x(img)
|
| 68 |
|
| 69 |
|
| 70 |
demo = gr.Interface(
|