Update app.py

app.py

@@ -20,6 +20,7 @@ logging.getLogger("http").setLevel(logging.WARNING)
 logging.getLogger("httpx").setLevel(logging.WARNING)
 
 import gradio as gr
+import spaces
 
 from conversation import default_conversation, conv_templates, SeparatorStyle
 
@@ -239,7 +240,7 @@ def stream_response(model, inputs, streamer, prompt, gen_kwargs):
         yield generated_text
 
 
-# @spaces.GPU
+@spaces.GPU
 def http_chat_bot(state, temperature, top_k, top_p, max_new_tokens):
     global model, args, streamer  # Use global model and args
     logging.info("http_chat_bot.")
@@ -317,6 +318,7 @@ def http_chat_bot(state, temperature, top_k, top_p, max_new_tokens):
     return (state, state.to_gradio_chatbot()) + (enable_btn,) * 2
 
 
+@spaces.GPU
 def http_gen_edit_bot(state, temperature, top_k, top_p, image_gen_temperature,
                       image_gen_top_k, image_gen_top_p, max_output_tokens,
                       llm_cfg_scale, resolution_wh, use_diffusion, diffusion_cfg_scale, diffusion_num_inference_steps):