llmahmad commited on
Commit
e34a9a0
·
verified ·
1 Parent(s): c9d9455

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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)