Update README.md

README.md

@@ -13,9 +13,9 @@ widget:
 
 
 ## Model description
-The Cuscuz-com-gemma 2b is a model derived from a fine tuning of the google/gemma-2b-it. This model was tuned to be specialized in the Northeast region of Brazil. 
-The model was specialized in a dataset that covered historical, geographical, economic, cultural and culinary issues in the northeast region.
-To make better use of the Cuscuz, the ideal is to use the model without quantization. This model is a small version of Cuscuz-7b whit another arquiteture
+The gemma-portuguese-2b model is a model trained with the alpaca dataset with the addition of other sets. 
+The model is mainly focused on text generation and instruction.
+The model was not trained on math and code tasks.
 
 ## How to Use
 
@@ -24,7 +24,7 @@ To make better use of the Cuscuz, the ideal is to use the model without quantiza
 from transformers import AutoTokenizer, pipeline
 import torch
 
-model = "rhaymison/cuscuz-com-gemma-2b"
+model = "rhaymison/gemma-portuguese-2b-it"
 
 tokenizer = AutoTokenizer.from_pretrained(model)
 pipeline = pipeline(
@@ -35,7 +35,7 @@ pipeline = pipeline(
 )
 
 messages = [
-    {"role": "user", "content": "Me conte sobre o estado de Sergipe."},
+    {"role": "user", "content": "Me conte sobre a ida do homem a Lua."},
 ]
 prompt = pipeline.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
 outputs = pipeline(
@@ -48,17 +48,17 @@ outputs = pipeline(
 )
 print(outputs[0]["generated_text"][len(prompt):].replace("model",""))
 
-#Sergipe é o menor estado do Nordeste brasileiro em extensão territorial. 
-#O estado de Sergipe é conhecido por suas praias, sua culinária à base de frutos do mar e sua importância histórica na produção de açúcar. 
-#Sergipe teve papel fundamental na produção de açúcar no Brasil colonial, sendo uma das regiões onde se concentraram os engenhos de açúcar.
+#A viagem à Lua foi um esforço monumental realizado pela Agência Espacial dos EUA entre 1969 e 1972.
+#Foi um marco significativo na exploração espacial e na ciência humana.
+#Aqui está uma visão geral de sua jornada: 1. O primeiro voo espacial humano foi o de Yuri Gagarin, que voou a Terra em 12 de abril de 1961.
 ```
 
 
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer2 = AutoTokenizer.from_pretrained("cuscuz-com-gemma-2b")
-model2 = AutoModelForCausalLM.from_pretrained("cuscuz-com-gemma-2b", device_map={"":0})
+tokenizer2 = AutoTokenizer.from_pretrained("rhaymison/gemma-portuguese-2b-it")
+model2 = AutoModelForCausalLM.from_pretrained("rhaymison/gemma-portuguese-2b-it", device_map={"":0})
 tokenizer2.pad_token = tokenizer2.eos_token
 tokenizer2.add_eos_token = True
 tokenizer2.add_bos_token, tokenizer2.add_eos_token
@@ -68,10 +68,7 @@ tokenizer2.padding_side = "right"
 ```python
 
 text = f"""
-Você é um assistente especialista em história do Nordeste Brasileiro.
-Você sempre responde de forma clara e educada e sempre com informações
-verdadeiras. Responda com detalhes e riquesas de informação
-<start_of_turn>Me conte sobre o Folclore Nordestino?<end_of_turn>
+<start_of_turn>Me conte sobre a ida do homem a Lua.<end_of_turn>
 <start_of_turn>model"""
 
 device = "cuda:0"
@@ -83,8 +80,9 @@ outputs = model2.generate(**inputs, max_new_tokens=100, do_sample=False)
 output = tokenizer2.decode(outputs[0], skip_special_tokens=True, skip_prompt=True)
 print(output.replace("model"," "))
 
-#O Folclore Nordestino é uma parte importante da nossa cultura, com manifestações como o bumba meu boi, o reisado, o maracatu e o repente.
-#Essa história é rica em lendas, contarorias e tradições que são passadas de geração em geração.  
+#A viagem à Lua foi um esforço monumental realizado pela Agência Espacial dos EUA entre 1969 e 1972.
+#Foi um marco significativo na exploração espacial e na ciência humana.
+#Aqui está uma visão geral de sua jornada: 1. O primeiro voo espacial humano foi o de Yuri Gagarin, que voou a Terra em 12 de abril de 1961. 
 ```
 
 ### Comments