Update app.py
Browse files
app.py
CHANGED
|
@@ -17,6 +17,7 @@ import matplotlib.pyplot as plt
|
|
| 17 |
from matplotlib.ticker import MaxNLocator
|
| 18 |
import gradio as gr
|
| 19 |
import tempfile
|
|
|
|
| 20 |
|
| 21 |
# Initialize models and other global variables
|
| 22 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
|
|
| 17 |
from matplotlib.ticker import MaxNLocator
|
| 18 |
import gradio as gr
|
| 19 |
import tempfile
|
| 20 |
+
import shutil
|
| 21 |
|
| 22 |
# Initialize models and other global variables
|
| 23 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|