Commit
•
bceac7d
1
Parent(s):
4e3af87
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,8 @@ import torch
|
|
| 5 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 6 |
|
| 7 |
# Inference with the pre-trained model
|
| 8 |
-
checkpoint1 = "microsoft/git-base"
|
|
|
|
| 9 |
processor = AutoProcessor.from_pretrained(checkpoint1)
|
| 10 |
#model1 = AutoModelForCausalLM.from_pretrained(checkpoint1, use_auth_token=access_token)
|
| 11 |
model1 = AutoModelForCausalLM.from_pretrained(checkpoint1)
|
|
|
|
| 5 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 6 |
|
| 7 |
# Inference with the pre-trained model
|
| 8 |
+
#checkpoint1 = "microsoft/git-base"
|
| 9 |
+
checkpoint1 = "vinid/plip"
|
| 10 |
processor = AutoProcessor.from_pretrained(checkpoint1)
|
| 11 |
#model1 = AutoModelForCausalLM.from_pretrained(checkpoint1, use_auth_token=access_token)
|
| 12 |
model1 = AutoModelForCausalLM.from_pretrained(checkpoint1)
|