Spaces:
Running on Zero
Running on Zero
[Admin maintenance] Migrate grant to ZeroGPU
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import argparse
|
| 2 |
|
| 3 |
import gradio as gr
|
|
@@ -62,7 +63,7 @@ def load_img (filename, norm=True,):
|
|
| 62 |
img = img.astype(np.float32)
|
| 63 |
return img
|
| 64 |
|
| 65 |
-
|
| 66 |
def process_img (image, prompt):
|
| 67 |
img = np.array(image)
|
| 68 |
img = img / 255.
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import argparse
|
| 3 |
|
| 4 |
import gradio as gr
|
|
|
|
| 63 |
img = img.astype(np.float32)
|
| 64 |
return img
|
| 65 |
|
| 66 |
+
@spaces.GPU
|
| 67 |
def process_img (image, prompt):
|
| 68 |
img = np.array(image)
|
| 69 |
img = img / 255.
|