Update app.py

app.py

@@ -4,6 +4,7 @@ import torch
 from PIL import Image
 import matplotlib.pyplot as plt
 from transformers import pipeline
+from typing import List, Dict, Union
 import warnings
 from io import BytesIO
 import importlib.util