Update app.py

app.py

@@ -5,6 +5,7 @@ from datetime import datetime
 import yaml
 import logging
 import gradio as gr
+from transformers import pipeline, AutoTokenizer
 
 # Create a directory for logs if it doesn't exist
 log_dir = "logs"