Update app.py

app.py

@@ -1,9 +1,18 @@
 import gradio as gr
+
+import argparse
+import cv2
+import imageio
+import math
+from math import ceil
+import matplotlib.pyplot as plt
+import matplotlib.animation as animation
 import numpy as np
+from PIL import Image
 import torch
-from torch import nn
-import imageio
-import cv2
+import torch.nn as nn
+import torch.nn.functional as F
+from torch.autograd import Function
 
 
 class RelationModuleMultiScale(torch.nn.Module):