update app

app.py

@@ -13,7 +13,7 @@ with gr.Blocks() as demo:
     """
     )
     image = gr.Image(type="pil")
-    with gr.Row():
+    with gr.Column(scale=1, min_width=600):
         # 設定輸出元件
         heights = gr.Textbox(label="Heignt")
         bust = gr.Textbox(label="Bust")

predictor.py

@@ -198,10 +198,10 @@ def evaluation(model, epoch, device, dataloaders):
             print(preds)
 
 def inference(inp_img, classes = ['big', 'small'], epoch = 6):
-    device = torch.device("cpu")
+    device = torch.device("cuda")
     translator= Translator(to_lang="zh-TW")
 
-    model = model = CUPredictor()
+    model = CUPredictor().to(device)
     model.load_state_dict(torch.load(f'models/model_{epoch}.pt'))
     # load image-to-text model
     processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")