Spaces:

ashwml
/

capstone_prometheus

Sleeping

ashwml commited on Dec 11, 2023

Commit

f2b0e17

1 Parent(s): d2864bb

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"
-vitgpt_model.to(device)
 def generate_caption(processor, model, image, tokenizer=None):
   # max_length = 16

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