Update app.py

app.py

@@ -2,6 +2,7 @@
 import torch
 from transformers import AutoTokenizer, EncoderDecoderModel
 import gradio as gr
+from spaces import GPU
 
 # デバイス設定 (Spacesのハードウェア設定に依存)
 # SpacesでGPUを利用する場合、自動的にCUDAが利用可能になります
@@ -45,6 +46,7 @@ except Exception as e:
     raise
 
 # --- Docstring生成関数 ---
+@GPU
 def generate_docstring(code: str) -> str:
     print("Received code snippet for docstring generation.") # 関数呼び出しログ
     if not code: