Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
6a3feb9
1
Parent(s):
31319dc
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ tokenizer, llava_model, image_processor, context_len = load_pretrained_model(
|
|
| 50 |
args.load_4bit,
|
| 51 |
device=args.device)
|
| 52 |
|
| 53 |
-
|
| 54 |
def bot_streaming(message, history):
|
| 55 |
print ("triggered")
|
| 56 |
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)
|