Manasa1 commited on
Commit
cfc8102
·
verified ·
1 Parent(s): 22a0b97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -1,10 +1,5 @@
1
  import gradio as gr
2
- from transformers import AutoModelForCausalLM, AutoTokenizer
3
- import random
4
- from datetime import datetime
5
- from PyPDF2 import PdfReader
6
- import json
7
- from transformers import AutoModelForCausalLM, AutoTokenizer
8
 
9
  # Replace 'username/your_model_name' with your Hugging Face model name
10
  model_dir = "Manasa1/your_model_name"
 
1
  import gradio as gr
2
+ from transformers import GPT2LMHeadModel, GPT2Tokenizer, pipeline
 
 
 
 
 
3
 
4
  # Replace 'username/your_model_name' with your Hugging Face model name
5
  model_dir = "Manasa1/your_model_name"