ginipick commited on
Commit
997f240
ยท
verified ยท
1 Parent(s): 2740e12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  import os
2
  import cv2
3
  import numpy as np
@@ -10,7 +14,7 @@ from typing import Tuple, Optional
10
 
11
  from PIL import Image
12
  from gradio_imageslider import ImageSlider
13
- from transformers import AutoModelForImageSegmentation, PreTrainedModel
14
  from torchvision import transforms
15
 
16
  import requests
@@ -18,9 +22,6 @@ from io import BytesIO
18
  import zipfile
19
  import random
20
 
21
- # ์˜ค๋ฒ„๋ผ์ด๋“œ: tie_weights()๋ฅผ ๋นˆ ํ•จ์ˆ˜๋กœ ์žฌ์ •์˜ํ•˜์—ฌ tie_weights ํ˜ธ์ถœ์„ ๋ฌด์‹œํ•จ.
22
- PreTrainedModel.tie_weights = lambda self: None
23
-
24
  torch.set_float32_matmul_precision('high')
25
  torch.jit.script = lambda f: f
26
 
 
1
+ # ๊ฐ€์žฅ ์ƒ๋‹จ์— ํŒจ์น˜๋ฅผ ์ ์šฉ (๋ชจ๋ธ ๋กœ๋”ฉ ์ „์— ์‹คํ–‰)
2
+ from transformers import PreTrainedModel
3
+ PreTrainedModel.tie_weights = lambda self: None
4
+
5
  import os
6
  import cv2
7
  import numpy as np
 
14
 
15
  from PIL import Image
16
  from gradio_imageslider import ImageSlider
17
+ from transformers import AutoModelForImageSegmentation
18
  from torchvision import transforms
19
 
20
  import requests
 
22
  import zipfile
23
  import random
24
 
 
 
 
25
  torch.set_float32_matmul_precision('high')
26
  torch.jit.script = lambda f: f
27