Update app.py

app.py

@@ -3,8 +3,8 @@ import torch
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 from diffusers import StableDiffusionPipeline
 from nltk.corpus import wordnet
-
-
+import nltk
+nltk.download('wordnet')
 
 def generate_text(prompt, temperature=0.7, top_k=50, repetition_penalty=1.2, max_length=None, min_length=10):
     text_model = "gpt2"