change model

app.py

@@ -71,12 +71,10 @@ class StrongGenerator(nn.Module):
 generator = StrongGenerator().to(device)
 
 
-# 載入訓練好的 Generator 模型（此處以第 10 個 epoch 為例，請根據實際情況修改）
-generator.load_state_dict(torch.load("./generator_epoch_10.pth", map_location=device))
+generator.load_state_dict(torch.load("./generator_epoch_70.pth", map_location=device))
 generator.eval()
 
 def restore_image(mosaic_image):
-    # 與訓練時相同的圖像轉換
     transform_in = transforms.Compose([
         transforms.Resize((256, 256)),
         transforms.ToTensor(),

generator_epoch_10.pth → generator_epoch_70.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:86ea6b42db1e7080a4bd2eec7656b101075b13535d7f249073cf62d025daf0f8
+oid sha256:7622d51270be899f993f9999e7f402c63cf6dfbdb6c37525c3bc8ddd67294b02
 size 31344538