Update app.py

app.py

@@ -6,7 +6,7 @@ import torch
 from PIL import Image
 from transformers import AutoProcessor, LlavaForConditionalGeneration, TextIteratorStreamer, TextStreamer
 
-# import spaces
+import spaces
 import argparse
 
 from llava_llama3.model.builder import load_pretrained_model
@@ -48,6 +48,7 @@ tokenizer, llava_model, image_processor, context_len = load_pretrained_model(
     args.load_4bit, 
     device=args.device)
 
+@spaces.GPU
 def bot_streaming(message, history):
     print(message)
     image_file = None