Upload 4 files

GPT2Tokenizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e7bb2fcbc544bd50ccbeca14449092ef286d44cd0da0cc8a09b2d330d6ffe714
+size 3095163

app.py

@@ -0,0 +1,14 @@
+import torch
+import gradio as gr
+
+
+model = torch.load("GPT2Model.pt")
+tokenizer = torch.load("GPT2Tokenizer.pt")
+def gpt2(prompt):
+    input_ids = tokenizer(prompt, return_tensors="pt").input_ids
+    outputs = model.generate(input_ids, do_sample=False, max_length=30)
+    return tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
+
+demo = gr.Interface(fn=gpt2, inputs="text", outputs="text")
+
+demo.launch()   

requiremens.txt

@@ -0,0 +1 @@
+torch