Update README.md

README.md

@@ -6,11 +6,11 @@ metrics:
 - accuracy
 ---
 
-# Model Card: POLLCHECK/paligemma
+# Model Card: POLLCHECK/Paligemma-bias-classifier
 
 ## Model Details
 
-**Model Name:** POLLCHECK/paligemma
+**Model Name:** POLLCHECK/Paligemma-bias-classifier
 
 **Model Description:** This is a fine-tuned PaliGemma model for news classification e.g. "biased" or "unbiased". In this particular task, the term 'biased' represents disinformation, propaganda, loaded language, negative associations, generalization, harm, hatred, satire 
 whereas 'unbiased' represents real news without the spread of misinformation, disinformation, and propaganda. The model can be used to identify potential bias in text and images, which is useful for applications in media analysis, content moderation, and research on bias in written communication.
@@ -50,7 +50,7 @@ from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
 device = "cuda:0"
 dtype = torch.bfloat16
 # Load the fine-tuned model and tokenizer
-model_id = "POLLCHECK/paligemma"  # path of the model
+model_id = "POLLCHECK/Paligemma-bias-classifier"  # path of the model
 model = PaliGemmaForConditionalGeneration.from_pretrained(
     model_id,
     torch_dtype=dtype,