missed import torch

app.py

@@ -9,6 +9,8 @@ from transformers import DetrFeatureExtractor, DetrForSegmentation
 from PIL import Image
 import gradio as gr
 import numpy as np
+import torch
+import torchvision
 
 # Returns a list with a color per ADE class (150 classes)
 # from https://huggingface.co/spaces/chansung/segformer-tf-transformers/blob/main/app.py
@@ -197,8 +199,8 @@ def predict_animal_mask(image):
 gr.Interface(predict_animal_mask, 
     inputs = input, 
     outputs = output,
-    title = 'Animals segmentation in images',
-    description = "An animal segmentation image webapp using DETR (End-to-End Object Detection) model with ResNet-50 backbone").launch()
+    title = 'Animals* segmentation in images',
+    description = "An animal* segmentation image webapp using DETR (End-to-End Object Detection) model with ResNet-50 backbone").launch()
 
 
 ####################################