working locally

app.py

@@ -33,7 +33,6 @@ def server(input: Inputs, output: Outputs, session: Session):
         
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         model2 = AutoModel.from_pretrained("ansal/sidewalk-segment")
-        model2.load_state_dict(torch.load('model.pth', map_location=device))
         model2.eval()
         model2.to(device)
 
@@ -62,13 +61,6 @@ def server(input: Inputs, output: Outputs, session: Session):
 
         combined = Image.alpha_composite(image, mask)
 
-        combined_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
-        original_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
-        image.save(original_file.name, "PNG", quality=100)
-        combined.save(combined_file.name, "PNG", quality=100)
-        
-        return original_file.name, combined_file.name
-    
         combined_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
         original_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
         image.save(original_file.name, "PNG", quality=100)