Update app.py

app.py

@@ -13,7 +13,7 @@ import cv2
 from matplotlib import pyplot as plt
 from inpainting import StableDiffusionInpaintingPipeline
 from torchvision import transforms
-#from clipseg.models.clipseg import CLIPDensePredT
+from clipseg.models.clipseg import CLIPDensePredT
 
 #from huggingface_hub import hf_hub_download
 #hf_hub_download(repo_id="ThereforeGames/txt2mask", filename="/repositories/clipseg/")
@@ -22,7 +22,7 @@ from torchvision import transforms
 
 #from huggingface_hub import Repository
 #with Repository(local_dir="clipseg", clone_from="ThereforeGames/txt2mask/repositories/clipseg/")
-
+"""
 import sys
 import os
 
@@ -31,7 +31,7 @@ zf = ZipFile('clipseg-master.zip', 'r')
 zf.extractall('./clipseg')
 zf.close()
 
-"""
+
 from huggingface_hub import HfApi
 api = HfApi()
 api.upload_folder(
@@ -63,7 +63,7 @@ pipe = StableDiffusionInpaintingPipeline.from_pretrained(
 
 model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64)
 model.eval()
-model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device('cuda')), strict=False)
+model.load_state_dict(torch.load('./clipseg-master/weights/rd64-uni.pth', map_location=torch.device('cuda')), strict=False)
 
 transform = transforms.Compose([
       transforms.ToTensor(),