hackergeek98 commited on
Commit
d2546a9
·
verified ·
1 Parent(s): 72c0741

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
  # Load your fine-tuned GPT-2 model from Hugging Face
5
- MODEL_NAME = "hackergeek98/therapist01" # Replace with your
6
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
7
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
8
 
 
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
  # Load your fine-tuned GPT-2 model from Hugging Face
5
+ MODEL_NAME = "hackergeek98/therapist01" # Replace w
6
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
7
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
8