soham-zero commited on
Commit
8ebafab
·
1 Parent(s): cc4a7cb
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -153,6 +153,7 @@ textarea:focus, input[type="text"]:focus {
153
  """
154
 
155
 
 
156
  def predict(prompt, opt_a, opt_b, opt_c, opt_d, opt_e):
157
  """Tokenize, run model forward pass, and return ranked answers."""
158
  if not prompt.strip():
 
153
  """
154
 
155
 
156
+ @spaces.GPU
157
  def predict(prompt, opt_a, opt_b, opt_c, opt_d, opt_e):
158
  """Tokenize, run model forward pass, and return ranked answers."""
159
  if not prompt.strip():