Spaces:

ashwml
/

capstone_prometheus

Sleeping

ashwml commited on Dec 11, 2023

Commit

3e2a45d

1 Parent(s): f2b0e17

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ tokenizer = AutoTokenizer.from_pretrained("model")
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model.to(device)
 def generate_caption(processor, model, image, tokenizer=None):
   # max_length = 16

 device = "cuda" if torch.cuda.is_available() else "cpu"
+cap_model.to(device)
 def generate_caption(processor, model, image, tokenizer=None):
   # max_length = 16