Files changed (1) hide show
  1. app.py +11 -3
app.py CHANGED
@@ -12,11 +12,19 @@ word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt",
12
  word_list = word_list_dataset["train"]['text']
13
 
14
  is_gpu_busy = False
 
15
  def infer(prompt, negative, scale):
16
  global is_gpu_busy
17
- for filter in word_list:
18
- if re.search(rf"\b{filter}\b", prompt):
19
- raise gr.Error("Unsafe content found. Please try again with different prompts.")
 
 
 
 
 
 
 
20
 
21
  images = []
22
  url = os.getenv('JAX_BACKEND_URL')
 
12
  word_list = word_list_dataset["train"]['text']
13
 
14
  is_gpu_busy = False
15
+
16
  def infer(prompt, negative, scale):
17
  global is_gpu_busy
18
+ # Yasaklı kelime kontrolü kaldırıldı
19
+ # GPU'nun meşgul olup olmadığını kontrol eder
20
+ if is_gpu_busy:
21
+ raise gr.Error("GPU is currently busy. Please try again later.")
22
+ # Çıkarım işlemini burada gerçekleştirin
23
+ # Örneğin: result = model.infer(prompt, negative, scale)
24
+ result = "Inference result" # Yer tutucu sonuç
25
+ return result
26
+
27
+
28
 
29
  images = []
30
  url = os.getenv('JAX_BACKEND_URL')