thak123 commited on
Commit
320f206
1 Parent(s): 846b6b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -90,7 +90,8 @@ config = model.vision_encoder.config
90
  # https://huggingface.co/FFZG-cleopatra/M2SA/blob/main/model.safetensors
91
  sf_filename = hf_hub_download("FFZG-cleopatra/M2SA", filename="model.safetensors")
92
 
93
- load_model(model,"model.safetensors") # model.load_state_dict(torch.load(model_args.model_name_or_path+"-finetuned/pytorch_model.bin"))
 
94
 
95
 
96
  model = AutoModelForSequenceClassification.from_pretrained(
 
90
  # https://huggingface.co/FFZG-cleopatra/M2SA/blob/main/model.safetensors
91
  sf_filename = hf_hub_download("FFZG-cleopatra/M2SA", filename="model.safetensors")
92
 
93
+ load_model(model, sf_filename)
94
+ # model.load_state_dict(torch.load(model_args.model_name_or_path+"-finetuned/pytorch_model.bin"))
95
 
96
 
97
  model = AutoModelForSequenceClassification.from_pretrained(