Update app.py

app.py

@@ -17,6 +17,7 @@ import matplotlib.pyplot as plt
 from matplotlib.ticker import MaxNLocator
 import gradio as gr
 import tempfile
+import shutil
 
 # Initialize models and other global variables
 device = 'cuda' if torch.cuda.is_available() else 'cpu'