minhdang commited on
Commit
f2bc373
1 Parent(s): 613e90a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -198,6 +198,7 @@ def get_prompt(conv) -> str:
198
  return conv.get_prompt
199
 
200
  @spaces.GPU
 
201
  def predict(
202
  text,
203
  image,
 
198
  return conv.get_prompt
199
 
200
  @spaces.GPU
201
+ @wrap_gen_fn
202
  def predict(
203
  text,
204
  image,