AnonymousSub commited on
Commit
9f9dd2d
1 Parent(s): c81f4e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,10 +7,10 @@ title = "Ayurveda4U"
7
  description = "LLM-Powered Medical Chatbot that will answer all your health-related queries with the help of Ayurvedic texts ynder the hood!"
8
  examples = [["How can you cure common cold using Ayurveda?"], ["What is the Ayurvedic equivalent of Paracetamol?"]]
9
 
10
- model_path = 'microsoft/phi-1_5'#'microsoft/DialoGPT-large' #'microsoft/biogpt' #'microsoft/BioGPT-large' #microsoft/DialoGPT-large
11
 
12
  tokenizer = AutoTokenizer.from_pretrained(model_path)
13
- model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True)
14
 
15
 
16
  def predict(input, history=[]):
 
7
  description = "LLM-Powered Medical Chatbot that will answer all your health-related queries with the help of Ayurvedic texts ynder the hood!"
8
  examples = [["How can you cure common cold using Ayurveda?"], ["What is the Ayurvedic equivalent of Paracetamol?"]]
9
 
10
+ model_path = 'tloen/alpaca-lora-7b' #'microsoft/phi-1_5'#'microsoft/DialoGPT-large' #'microsoft/biogpt' #'microsoft/BioGPT-large' #microsoft/DialoGPT-large
11
 
12
  tokenizer = AutoTokenizer.from_pretrained(model_path)
13
+ model = AutoModelForCausalLM.from_pretrained(model_path)
14
 
15
 
16
  def predict(input, history=[]):