Spaces:

alexnasa
/

SuperResolution

Running on Zero

alexnasa commited on Jul 11

Commit

cf3b00f

verified ·

1 Parent(s): 6cc150a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -121,7 +121,7 @@ tag_model = ram(pretrained='preset/models/ram_swin_large_14m.pth',
 tag_model.eval()
 tag_model.to(device, dtype=weight_dtype)
-@torch.no_grad()
 def process(
     input_image: Image.Image,
     user_prompt: str,

 tag_model.eval()
 tag_model.to(device, dtype=weight_dtype)
+@spaces.GPU()
 def process(
     input_image: Image.Image,
     user_prompt: str,