Spaces:
Running
on
Zero
Running
on
Zero
jiminHuang
commited on
Commit
•
de95975
1
Parent(s):
6952c25
Update app.py
Browse files
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)
|