m

handler.py

@@ -999,48 +999,17 @@ class API_Starter:
         server = pywsgi.WSGIServer(('0.0.0.0', args.port), self.app)
         server.serve_forever()  
 
-def main_api(args):
-    if args.port == 0:
-        print('Please give valid port!')
-    else:
-        api_starter = API_Starter()
-        api_starter.run() 
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser("Grounded SAM demo", add_help=True)
-    parser.add_argument("--debug", action="store_true", help="using debug mode")
-    parser.add_argument("--share", action="store_true", help="share the app")
-    parser.add_argument("--port", "-p", type=int, default=7860, help="port")
-    args, _ = parser.parse_known_args()
-    print(f'args = {args}')
+device = set_device()
 
-    if os.environ.get('IS_MY_DEBUG') is None:
-        os.system("pip list")
-    
-    device = set_device()
-    if device == 'cpu':
-        kosmos_enable = False
+groundingdino_model = load_groundingdino_model('cuda:0')
+load_sam_model("cuda:0")
 
-    if kosmos_enable:
-        kosmos_model, kosmos_processor = load_kosmos_model(device)
-        
-    if groundingdino_enable:
-        groundingdino_model = load_groundingdino_model('cpu')
-    
-    if sam_enable:
-        load_sam_model(device)
+load_sd_model("cuda:0")
 
-    if inpainting_enable:
-        load_sd_model(device)
+load_lama_cleaner_model("cuda:0")
 
-    if lama_cleaner_enable:
-        load_lama_cleaner_model(device)
+load_ram_model("cuda:0")
 
-    if ram_enable:
-        load_ram_model(device)
-    
-    if os.environ.get('IS_MY_DEBUG') is None:
-        os.system("pip list")
 
 def just_fucking_get_sd_mask(input_pil, prompt):
     return run_anything_task(input_pil, prompt, "inpainting", "", 0.3, 0.25, 0.8, "merge", "type what to detect below", "segment", "10", 5, "Brief")