Update app.py

app.py

@@ -1,16 +1,13 @@
 import os
 import gradio as gr
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
-
+@spaces.GPU
 messages = [
     {"role": "user", "content": "Who are you?"},
 ]
 pipe = pipeline("text-generation", model="SakanaAI/DiscoPOP-zephyr-7b-gemma")
 pipe(messages)
 
-# CUDAを無効にする
-os.environ["CUDA_VISIBLE_DEVICES"] = ""
-
 # 日本語モデルを指定
 model_name = "SakanaAI/DiscoPOP-zephyr-7b-gemma"  # Sakana AIのモデル名を指定