temp

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 
 import torch
+
+import matplotlib.pyplot as plt
+
 from torchvision import transforms
 
 import fire_network