Update README.md

README.md

@@ -6,8 +6,22 @@ tags: []
 # Model Card for Model ID
 
 <!-- Provide a quick summary of what the model is/does. -->
-This is the baseline model for the news source classification project. To perform inference, you call model.predict('test.csv'). 
+This is the baseline model for the news source classification project. 
 
+Code to load the model:
+from huggingface_hub import hf_hub_download
+import joblib
+
+repo_id='awngsz/baseline_model'
+filename='CIS5190_Proj2_AWNGSZ.joblib'
+
+file_path=hf_hub_download(repo_id=repo_id, filename=filename)
+model=joblib.load(file_path)
+
+print(model)
+
+Code to perform inference: 
+model.predict('test.csv')
 
 ## Model Details