Update app.py

app.py

@@ -1,6 +1,12 @@
 import gradio as gr
 import transformers
 from transformers import AutoModelWithLMHead, AutoTokenizer, pipeline
+from transformers import GPT2Tokenizer, GPT2Model
+
+model = GPT2Model.from_pretrained('LorenzoDeMattei/GePpeTto')
+tokenizer = GPT2Tokenizer.from_pretrained(
+    'LorenzoDeMattei/GePpeTto',
+)
 
 shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('LorenzoDeMattei/GePpeTto'),
     tokenizer='LorenzoDeMattei/GePpeTto',