Puttipong13 commited on
Commit
94d2dcd
·
verified ·
1 Parent(s): 3835087

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -8,6 +8,18 @@ import sys
8
  sys.modules["triton"] = None # triton segfault ตอน import บน CPU-only Space: ทำให้ import ล้มเหลวแบบปกติ torch จะข้ามไปเอง
9
  import faulthandler
10
  faulthandler.enable() # ถ้า crash อีก จะพิมพ์ตำแหน่งที่พังลง log
 
 
 
 
 
 
 
 
 
 
 
 
11
  import torch # import torch ก่อน tensorflow เพื่อเลี่ยง segfault
12
  # ---------------------------------------------------------------------
13
 
 
8
  sys.modules["triton"] = None # triton segfault ตอน import บน CPU-only Space: ทำให้ import ล้มเหลวแบบปกติ torch จะข้ามไปเอง
9
  import faulthandler
10
  faulthandler.enable() # ถ้า crash อีก จะพิมพ์ตำแหน่งที่พังลง log
11
+
12
+ # --- ZeroGPU: ต้อง import spaces ก่อน torch และต้องมีฟังก์ชัน @spaces.GPU อย่างน้อย 1 ตัว ---
13
+ try:
14
+ import spaces
15
+ @spaces.GPU
16
+ def _zerogpu_placeholder():
17
+ """ฟังก์ชันเปล่า ใช้ผ่านเงื่อนไข ZeroGPU เท่านั้น (โมเดล TensorFlow ของเรารันบน CPU ตามปกติ)"""
18
+ return None
19
+ except ImportError:
20
+ pass # รันนอก Hugging Face Spaces ไม่ต้องใช้
21
+ # ---------------------------------------------------------------------------------------
22
+
23
  import torch # import torch ก่อน tensorflow เพื่อเลี่ยง segfault
24
  # ---------------------------------------------------------------------
25