Update app.py

app.py

@@ -1,3 +1,4 @@
+import gradio as gr
 import torch
 import torch.nn as nn
 import torchvision.transforms as transforms
@@ -92,4 +93,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()