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
 
@@ -20,7 +20,7 @@ if torch.cuda.is_available():
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 
-# @spaces.GPU
+@spaces.GPU
 def generate(
  message: str,
  chat_history: list[tuple[str, str]],