Update README.md

README.md

@@ -1,56 +1,53 @@
----
-tags:
-- medical
-license: mit
-datasets:
-- Mohammed-Altaf/medical-instruction-120k
-language:
-- en
-library_name: transformers
----
-Please note that the chatbot is designed for research purposes only and is not intended for use in real medical settings. While the chatbot has been trained to provide accurate and helpful responses, it is not a substitute for professional medical advice, diagnosis, or treatment. The information provided by the chatbot should not be used to make medical decisions, and any health concerns should be addressed by a licensed healthcare provider.
-
-## Quickstart
+
+
+## Aviso Legal
+
+Este chatbot foi projetado apenas para fins de pesquisa e não se destina ao uso em configurações médicas reais. Embora o chatbot tenha sido treinado para fornecer respostas precisas e úteis, ele não substitui o aconselhamento médico profissional, o diagnóstico ou o tratamento. As informações fornecidas pelo chatbot não devem ser usadas para tomar decisões médicas, e quaisquer preocupações com a saúde devem ser tratadas por um profissional de saúde licenciado.
+
+## Início Rápido
 
 ```python
 import torch
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
 
-path = "Mohammed-Altaf/Medical-ChatBot"
-device = "cuda" if torch.cuda.is_available() else "cpu"
-tokenizer = GPT2Tokenizer.from_pretrained(path)
-model = GPT2LMHeadModel.from_pretrained(path).to(device)
+caminho = "Mohammed-Altaf/Medical-ChatBot"
+dispositivo = "cuda" if torch.cuda.is_available() else "cpu"
+tokenizer = GPT2Tokenizer.from_pretrained(caminho)
+modelo = GPT2LMHeadModel.from_pretrained(caminho).to(dispositivo)
 
-prompt_input = (
-    "The conversation between human and AI assistant.\n"
-    "[|Human|] {input}\n"
-    "[|AI|]"
+entrada_prompt = (
+    "A conversa entre humano e assistente de IA.\n"
+    "[|Humano|] {input}\n"
+    "[|IA|]"
 )
-sentence = prompt_input.format_map({'input': "what is parkinson's disease?"})
-inputs = tokenizer(sentence, return_tensors="pt").to(device)
+frase = entrada_prompt.format_map({'input': "o que é a doença de Parkinson?"})
+inputs = tokenizer(frase, return_tensors="pt").to(dispositivo)
 
 with torch.no_grad():
-    beam_output = model.generate(**inputs,
-                                min_new_tokens=1, 
-                                max_length=512,
-                                num_beams=3,
-                                repetition_penalty=1.2,
-                                early_stopping=True,
-                                eos_token_id=198 
-                                )
+    beam_output = modelo.generate(**inputs,
+                                  min_new_tokens=1, 
+                                  max_length=512,
+                                  num_beams=3,
+                                  repetition_penalty=1.2,
+                                  early_stopping=True,
+                                  eos_token_id=198 
+                                  )
     print(tokenizer.decode(beam_output[0], skip_special_tokens=True))
 ```
 
-## Example Outputs
+## Exemplos de Saída
+
 ```
-The conversation between human and AI assistant.
-[|Human|] what is parkinson's disease?
-[|AI|] Parkinson's disease is a neurodegenerative disorder that affects movement. It is caused by the loss of dopamine-producing cells in the brain.
+A conversa entre humano e assistente de IA.
+[|Humano|] o que é a doença de Parkinson?
+[|IA|] A doença de Parkinson é um distúrbio neurodegenerativo que afeta o movimento. É causada pela perda de células produtoras de dopamina no cérebro.
 ```
 
 ```
-The conversation between human and AI assistant.
-[|Human|] what type of honey is best for a bad covid cough?
-[|AI|] Manuka honey has been shown to have anti-inflammatory and antibacterial properties that can help alleviate symptoms of a bad covid cough.
-```
+A conversa entre humano e assistente de IA.
+[|Humano|] que tipo de mel é melhor para uma tosse ruim de covid?
+[|IA|] O mel de Manuka mostrou ter propriedades anti-inflamatórias e antibacterianas que podem ajudar a aliviar os sintomas de uma tosse ruim de covid.
+```
+
+---