Update app.py

app.py

@@ -1,9 +1,20 @@
-#import nltk
+import os
+try:
+  import nltk
+except ModuleNotFoundError:
+  os.system("pip3 install nltk")
+try:
+  import transformers
+except ModuleNotFoundError:
+  os.system("pip3 install transformers")
+
+
+import nltk
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 """ Created by Riky Ripaldo """
-#nltk.download('punkt')
+nltk.download('punkt')
 
 tokenz = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
 modelz = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")