234bcn commited on
Commit
aae0613
·
1 Parent(s): da63f67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -1,3 +1,6 @@
1
- import gradio as gr
2
-
3
- gr.Interface.load("models/gpt2").launch()
 
 
 
 
1
+ from transformers import GPT2Tokenizer, GPT2Model
2
+ tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
3
+ model = GPT2Model.from_pretrained('gpt2')
4
+ text = "Replace me by any text you'd like."
5
+ encoded_input = tokenizer(text, return_tensors='pt')
6
+ output = model(**encoded_input)