ahmed-7124 commited on
Commit
354b2ce
Β·
verified Β·
1 Parent(s): 67ce37f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
  # Load the fine-tuned model and tokenizer
5
- tokenizer = AutoTokenizer.from_pretrained("harishussain12/PastelMed")
6
- model = AutoModelForCausalLM.from_pretrained("harishussain12/PastelMed")
7
 
8
  # Function to generate response from the model
9
  def doctor_consultant(query):
 
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
  # Load the fine-tuned model and tokenizer
5
+ tokenizer = AutoTokenizer.from_pretrained("ahmed-7124/dgptAW")
6
+ model = AutoModelForCausalLM.from_pretrained("ahmed-7124/dgptAW")
7
 
8
  # Function to generate response from the model
9
  def doctor_consultant(query):