app.py

import streamlit as st
import torch
from transformers import AutoModelWithLMHead, AutoTokenizer

# Title and Description
st.title("Text Generation with Huggingface Model")
st.write("This program uses a huggingface model to generate text.")

# Tokenizer and Model
tokenizer = AutoTokenizer.from_pretrained("gpt2")
model = AutoModelWithLMHead.from_pretrained("gpt2")