Update app.py

app.py

@@ -1,13 +1,18 @@
+import os
 import tempfile
-from PIL import Image
 from pathlib import Path
-from shiny import App, Inputs, Outputs, Session, reactive, render, ui
-from shiny.types import FileInfo
+import pandas as pd
+from PIL import Image
 import torch
 import numpy as np
-import os
-from transformers import SamModel
 import torchvision.transforms as transforms
+import matplotlib.pyplot as plt
+from shiny import App, Inputs, Outputs, Session, reactive, render, ui
+from shiny.types import FileInfo
+
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers_cache/"
+
+from transformers import SamModel
 
 image_resize_transform = transforms.Compose([
     transforms.Resize((1024, 1024)), 
@@ -37,12 +42,10 @@ def server(input: Inputs, output: Outputs, session: Session):
         transform = image_resize_transform
         image_tensor = transform(image).to(device)
 
-        
         with torch.no_grad():
-            outputs = model2(pixel_values=image_tensor.unsqueeze(0),multimask_output=False)
+            outputs = model2(pixel_values=image_tensor.unsqueeze(0), multimask_output=False)
         predicted_masks = outputs.pred_masks.squeeze(1) 
         predicted_masks = predicted_masks[:, 0, :, :]
-
         
         mask_tensor = predicted_masks.cpu().detach().squeeze()
         mask_array = mask_tensor.numpy()  
@@ -50,19 +53,16 @@ def server(input: Inputs, output: Outputs, session: Session):
         mask = Image.fromarray(mask_array)
         mask = mask.resize((1024, 1024), Image.LANCZOS)
         mask = mask.convert('RGBA')
-
         
         alpha = Image.new('L', mask.size, 128)
         mask.putalpha(alpha)
 
-        
         image = Image.open(file[0]["datapath"]).convert('RGB')  
         image = image.resize((1024, 1024), Image.LANCZOS)  
         image = image.convert('RGBA')
 
         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)