2 commit

app.py

@@ -8,10 +8,10 @@ from quantization import OnnxSetFitModel
 from transformers import AutoTokenizer
 # Load the models
 
-# model1 = SetFitModel.from_pretrained("hsmashiana/base_model_hpml")
-# ort_model = ORTModelForFeatureExtraction.from_pretrained("hsmashiana/optimized_model_hpml", file_name="model_quantized.onnx")
-# tokenizer = AutoTokenizer.from_pretrained("hsmashiana/optimized_model_hpml")
-# model3 = OnnxSetFitModel(ort_model, tokenizer, model1.model_head)
+model1 = SetFitModel.from_pretrained("hsmashiana/base_model_hpml")
+ort_model = ORTModelForFeatureExtraction.from_pretrained("hsmashiana/optimized_model_hpml", file_name="model_quantized.onnx")
+tokenizer = AutoTokenizer.from_pretrained("hsmashiana/optimized_model_hpml")
+model3 = OnnxSetFitModel(ort_model, tokenizer, model1.model_head)
 
 decode = {0:"World",1:"Sports",2:"Business",3:"Sci/Tech"}