Update README.md

README.md

@@ -45,7 +45,7 @@ import requests, json
 from time import sleep
 from tqdm.auto import tqdm, trange
 
-API_URL = "https://api-inference.huggingface.co/models/jslin09/bloom-560m-finetuned-fraud"
+API_URL = "https://api-inference.huggingface.co/models/jslin09/gemma2-2b-fraud"
 API_TOKEN = 'XXXXXXXXXXXXXXX' # 調用模型的 API token
 headers = {"Authorization": f"Bearer {API_TOKEN}"} 
 
@@ -79,10 +79,12 @@ print(response[0]['generated_text'])
   <summary> 點擊後展開 </summary>
 <pre>
   <code>
+# Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("jslin09/bloom-560m-finetuned-fraud")
-model = AutoModelForCausalLM.from_pretrained("jslin09/bloom-560m-finetuned-fraud")
+tokenizer = AutoTokenizer.from_pretrained("jslin09/gemma2-2b-fraud")
+model = AutoModelForCausalLM.from_pretrained("jslin09/gemma2-2b-fraud")
+
 </code>
 </pre>
 </details>