Hugging Face's logo

Spaces:
RandomNameAnd6
/
DharGPT-Demo
Sleeping

App Files Community
RandomNameAnd6 commited on
Commit
b4c48e8
·
verified ·
1 Parent(s): bf07570

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel
8
 
9
 
10
  tokenizer = GPT2Tokenizer.from_pretrained("gpt2-medium")
11
- model = GPT2LMHeadModel.from_pretrained("DharGPT-Small")
12
 
13
  def generate_text(prompt):
14
  input_ids = tokenizer.encode(prompt, return_tensors="pt")
 
8
 
9
 
10
  tokenizer = GPT2Tokenizer.from_pretrained("gpt2-medium")
11
+ model = GPT2LMHeadModel.from_pretrained("RandomNameAnd6/DharGPT-Small")
12
 
13
  def generate_text(prompt):
14
  input_ids = tokenizer.encode(prompt, return_tensors="pt")