Update app.py

app.py

@@ -6,7 +6,7 @@ from typing import Iterator
 import torch
 
 import gradio as gr
-# import spaces
+import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 
@@ -21,7 +21,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 tokenizer.pad_token_id = tokenizer.eos_token_id
 
 
-# @spaces.GPU
+@spaces.GPU
 def generate(
     message: str,
     chat_history: list[tuple[str, str]],