llmModeluser commited on
Commit
200d90e
·
verified ·
1 Parent(s): f0ca391
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,11 +1,13 @@
1
  import gradio as gr
2
  from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
3
 
4
- # Load the pre-trained model and tokenizer from Hugging Face
5
  model_name = "llmModeluser/therapy_trained_model"
6
- tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForCausalLM.from_pretrained(model_name)
8
 
 
 
 
9
  # Create the text generation pipeline
10
  nlp = pipeline("text-generation", model=model, tokenizer=tokenizer)
11
 
 
1
  import gradio as gr
2
  from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
3
 
4
+ # Load the pre-trained model from Hugging Face
5
  model_name = "llmModeluser/therapy_trained_model"
 
6
  model = AutoModelForCausalLM.from_pretrained(model_name)
7
 
8
+ # Load the tokenizer directly from the model's file path
9
+ tokenizer = AutoTokenizer.from_pretrained("llmModeluser/therapy_trained_model")
10
+
11
  # Create the text generation pipeline
12
  nlp = pipeline("text-generation", model=model, tokenizer=tokenizer)
13