Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -121,7 +121,7 @@ tag_model = ram(pretrained='preset/models/ram_swin_large_14m.pth',
|
|
| 121 |
tag_model.eval()
|
| 122 |
tag_model.to(device, dtype=weight_dtype)
|
| 123 |
|
| 124 |
-
@spaces.GPU(
|
| 125 |
def process(
|
| 126 |
input_image: Image.Image,
|
| 127 |
user_prompt: str,
|
|
|
|
| 121 |
tag_model.eval()
|
| 122 |
tag_model.to(device, dtype=weight_dtype)
|
| 123 |
|
| 124 |
+
@spaces.GPU()
|
| 125 |
def process(
|
| 126 |
input_image: Image.Image,
|
| 127 |
user_prompt: str,
|