Update app.py

app.py

@@ -22,16 +22,14 @@ def visualize_segmentation(image, prompts, preds):
      
 
 def segment(img, clases):
+    print(img)
     prompts = clases.split(',')
 
     inputs = processor(text=prompts, images=[img] * len(img), padding="max_length", return_tensors="pt")
     
-    with torch.no_grad():
-      outputs = model(**inputs)
+    outputs = model(**inputs)
     preds = outputs.logits.unsqueeze(1)
 
-    
-    
     return "Hello " + prompts + "!!"
 
 demo = gr.Interface(fn=segment, inputs=["image","text"], outputs="text")