jeevana commited on
Commit
1bd5bd8
·
verified ·
1 Parent(s): e1b6445

Create app1.py

Browse files
Files changed (1) hide show
  1. app1.py +14 -0
app1.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import torch
3
+ from transformers import GPT2Tokenizer, GPT2LMHeadModel
4
+
5
+ checkpoint = "gpt2"
6
+ tokenizer = GPT2Tokenizer.from_pretrained(checkpoint)
7
+
8
+
9
+ # Load the fine-tuned model and tokenizer
10
+ my_model = GPT2LMHeadModel.from_pretrained("jeevana/GenerativeQnASystem")
11
+ my_tokenizer = GPT2Tokenizer.from_pretrained("jeevana/GenerativeQnASystem")
12
+
13
+
14
+