Manasa1 commited on
Commit
2f4a891
·
verified ·
1 Parent(s): b3ba6d3

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 pipeline, GPT2LMHeadModel, GPT2Tokenizer
3
 
4
  # Load the fine-tuned GPT-2 model and tokenizer
5
- model_dir = "Manasa1/model_name"
6
  fine_tuned_model = GPT2LMHeadModel.from_pretrained(model_dir)
7
  fine_tuned_tokenizer = GPT2Tokenizer.from_pretrained(model_dir)
8
 
 
2
  from transformers import pipeline, GPT2LMHeadModel, GPT2Tokenizer
3
 
4
  # Load the fine-tuned GPT-2 model and tokenizer
5
+ model_dir = "Manasa1/finetuned_GPT23"
6
  fine_tuned_model = GPT2LMHeadModel.from_pretrained(model_dir)
7
  fine_tuned_tokenizer = GPT2Tokenizer.from_pretrained(model_dir)
8