Update README.md

README.md

@@ -66,7 +66,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 hf_auth = 'HF_ACCESS_KEY'
 
 model_id = "recogna-nlp/Phi-Bode"
-gembode = AutoModelForCausalLM.from_pretrained(model_id, device_map={"":0}, trust_remote_code=True, token=hf_auth)
+model = AutoModelForCausalLM.from_pretrained(model_id, device_map={"":0}, trust_remote_code=True, token=hf_auth)
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True, token=hf_auth)
 
 def get_completion(model, tokenizer, query : str, input : str = '', device = 'cuda:0', max_new_tokens=128) -> str:
@@ -90,7 +90,7 @@ def get_completion(model, tokenizer, query : str, input : str = '', device = 'cu
     decoded = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
     return (decoded[len(prompt):])
 
-result = get_completion(model=gembode, tokenizer=tokenizer, query="Qual é a capital da França?")
+result = get_completion(model=model, tokenizer=tokenizer, query="Qual é a capital da França?")
 print(result)
 #Exemplo de resposta obtida: A capital da França é Paris. A cidade tem uma estratégia de transporte moderno difícil entre todos os lugares, incluindo ferroviário, busca, metro e línguações. Para obter uma avaliação completa da cidade, visita esta aumentar a experiência gastronômica, cultural e natural.
 ```