Disease Prognosis and Precautions Text2Text Generation
Welcome to the Disease Prognosis and Precautions Text2Text Generation repository! Fine-tuned microsoft/GODEL-v1_1-large-seq2seq. The model is designed to generate responses for disease prognosis and recommended precautions based on given symptoms.
Model Overview
The model in this repository is a text-to-text generation model. It takes a prompt in the form of symptoms related to a particular disease and generates a response that includes the potential disease prognosis along with recommended precautions. The columns used in the training dataset are:
- Disease: The name of the disease related to the symptoms.
- Symptoms: The list of symptoms provided in the prompt.
- Precautions: The recommended precautions for the identified disease.
Examples
Here are some examples of how you can use the model:
Example 1
Prompt: "I am feeling continuous sneezing, shivering and chills"
Response: "Seems like allergy. You should try to avoid dust and air pollution."
Example 2
Prompt: "I am feeling itching, skin rash and patches"
Response: "Seems like fungal infection. You should bathe twice a day and use antifungal soap."
How to Use
To use the model for generating disease prognosis and precautions based on symptoms, you can use the generate function provided by the Hugging Face Transformers library. Here's a basic example using Python:
1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3# Load the model and tokenizer
4model_name = "shanover/medbot_godel_v3"
5model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7
8# Define your symptom prompt
9prompt = "I am feeling continuous sneezing, shivering and chills"
10
11def generate_response(input_text, model, tokenizer, max_length):
12 input_ids = tokenizer.encode(input_text, return_tensors="pt", max_length=max_length, truncation=True)
13 input_ids = input_ids.to(device)
14
15 with torch.no_grad():
16 output_ids = model.generate(input_ids)
17
18 generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
19 return generated_text
20
21print(generate_response(prompt, model, tokenizer))
Remember to replace "shanover/medbot_godel_v3" with the actual name or path of the model you've downloaded or fine-tuned.
Acknowledgments
Issues and Contributions
If you encounter any issues while using the model or have suggestions for improvements, please feel free to open an issue in this repository. Contributions are also welcome!
Disclaimer
Please note that the information generated by the model is for informational purposes only and should not be considered a substitute for professional medical advice. Always consult a medical professional for accurate diagnoses and treatments.
Thank you for using the Disease Prognosis and Precautions Text2Text Generation model! We hope it proves to be a helpful tool.
