add model

App_main.py

@@ -2,7 +2,7 @@ from collections import OrderedDict
 
 import gradio as gr
 import os
-
+os.system('pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113')
 import torch
 from torchvision import transforms
 from torchvision.transforms import InterpolationMode
@@ -55,6 +55,7 @@ def seg_buildings(Image, Checkpoint):
     sample = construct_sample(Image, mean, std)
     model = build_model(checkpoint)
     device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
+    print('Use: ', device)
 
     model = model.to(device)
     model.eval()