Update app.py

app.py

@@ -108,7 +108,7 @@
 #     demo.launch()
 
 
-
+import os
 import torch
 from transformers import pipeline
 from PIL import Image