Update requirements and dependencies

app.py

@@ -1,6 +1,4 @@
 import io
-import os.path
-import sys
 
 import gradio as gr
 import matplotlib.pyplot as plt
@@ -8,13 +6,11 @@ import numpy as np
 import scipy.sparse
 import torch
 import torch.nn.functional as F
-import torchvision
 import torchvision.transforms.functional as TF
 from gradio.inputs import Image as GradioInputImage
 from gradio.outputs import Image as GradioOutputImage
 from PIL import Image
 from scipy.sparse.linalg import eigsh
-from skimage.color import label2rgb
 from torch.utils.hooks import RemovableHandle
 from torchvision import transforms
 from torchvision.utils import make_grid
@@ -193,7 +189,7 @@ examples = [f"examples/{stem}.jpg" for stem in [
 ]]
 
 title = "Deep Spectral Segmentation"
-description = "Deep spectral segmentation..."
+description = "Deep spectral segmentation (add description here) ..."
 thumbnail = "https://raw.githubusercontent.com/gradio-app/hub-echonet/master/thumbnail.png"
 
 # Gradio

requirements.txt

@@ -0,0 +1,6 @@
+-f https://download.pytorch.org/whl/torch_stable.html
+numpy
+matplotlib
+torch==1.9.1+cpu 
+torchvision==0.10.1+cpu
+scipy