oucgc1996 commited on
Commit
e789509
·
verified ·
1 Parent(s): 48636f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,6 +7,7 @@ import gradio as gr
7
  from gradio_rangeslider import RangeSlider
8
  import time
9
  import numba
 
10
 
11
  is_stopped = False
12
 
@@ -22,7 +23,7 @@ def stop_generation():
22
  is_stopped = True
23
  return "Generation stopped."
24
 
25
- @numba.jit(nopython=True)
26
  def generate_sequence(length, vocab_mlm, seq, new_seq, τ, input_text):
27
  for i in range(length):
28
  if is_stopped:
 
7
  from gradio_rangeslider import RangeSlider
8
  import time
9
  import numba
10
+ from numba import objmode
11
 
12
  is_stopped = False
13
 
 
23
  is_stopped = True
24
  return "Generation stopped."
25
 
26
+ @numba.jit(nopython=False)
27
  def generate_sequence(length, vocab_mlm, seq, new_seq, τ, input_text):
28
  for i in range(length):
29
  if is_stopped: