Spaces:

yfyangd
/

PictureBookUnderstanding

Runtime error

yfyangd commited on Sep 2, 2022

Commit

8c9ecba

1 Parent(s): 52ae2ad

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -71,7 +71,7 @@ def interrogate(cover):
     for model_name in models:
         #print(f"Interrogating with {model_name}...")
         model, preprocess = clip.load(model_name)
-        model.cuda().eval()
         images = preprocess(image).unsqueeze(0).cuda()
         with torch.no_grad():

     for model_name in models:
         #print(f"Interrogating with {model_name}...")
         model, preprocess = clip.load(model_name)
+        #model.cuda().eval()
         images = preprocess(image).unsqueeze(0).cuda()
         with torch.no_grad():