jiminHuang commited on
Commit
de95975
1 Parent(s): 6952c25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -50,6 +50,7 @@ tokenizer, llava_model, image_processor, context_len = load_pretrained_model(
50
  args.load_4bit,
51
  device=args.device)
52
 
 
53
  def bot_streaming(message, history):
54
  print ("triggered")
55
  print(message)
 
50
  args.load_4bit,
51
  device=args.device)
52
 
53
+ @spaces.gpu
54
  def bot_streaming(message, history):
55
  print ("triggered")
56
  print(message)