Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,9 @@
|
|
1 |
# Load the GPT-2 model and tokenizer
|
|
|
2 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
|
|
|
|
|
|
3 |
model_name = 'gpt2-large'
|
4 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
|
5 |
model = GPT2LMHeadModel.from_pretrained(model_name)
|
|
|
1 |
# Load the GPT-2 model and tokenizer
|
2 |
+
import streamlit as st
|
3 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
4 |
+
import torch
|
5 |
+
|
6 |
+
|
7 |
model_name = 'gpt2-large'
|
8 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
|
9 |
model = GPT2LMHeadModel.from_pretrained(model_name)
|