Spaces:

Futuretop
/

Image-to-Text

Runtime error

Futuretop commited on May 26

Commit

9876926

verified ·

1 Parent(s): cdf1664

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ subprocess.run(
     check=True
 )
-device = torch.device("cuda" if torch.cuda.is_available() else "CPU")
 config = AutoConfig.from_pretrained(model_id, trust_remote_code=True)

     check=True
 )
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 config = AutoConfig.from_pretrained(model_id, trust_remote_code=True)