Update app.py

app.py

@@ -18,6 +18,7 @@ import torch.nn.functional as F
 #from sklearn.decomposition import PCA
 import matplotlib.pyplot as plt
 import json
+import gradio as gr
 
 # Load configuration file 
 with open('config.json', 'r') as config_file: