ashwml commited on
Commit
5cb1c99
·
1 Parent(s): c845a40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -39,9 +39,11 @@ import torch
39
  # f1_metric.set(f1)
40
 
41
  feature_extractor = ViTImageProcessor.from_pretrained("model")
42
- print(feature_extractor)
43
  cap_model = VisionEncoderDecoderModel.from_pretrained("model")
 
44
  tokenizer = AutoTokenizer.from_pretrained("model")
 
45
 
46
  device = "cuda" if torch.cuda.is_available() else "cpu"
47
 
 
39
  # f1_metric.set(f1)
40
 
41
  feature_extractor = ViTImageProcessor.from_pretrained("model")
42
+ print("feature_extractor--",feature_extractor)
43
  cap_model = VisionEncoderDecoderModel.from_pretrained("model")
44
+ print("cap_model--",cap_model)
45
  tokenizer = AutoTokenizer.from_pretrained("model")
46
+ print("tokenizer--",tokenizer)
47
 
48
  device = "cuda" if torch.cuda.is_available() else "cpu"
49