Update app.py

app.py

@@ -1,3 +1,10 @@
+# app.py
+import os
+
+# 1. 在导入 PyTorch、Transformers 等任何使用 GPU 的库之前，先屏蔽 GPU
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
+
+import torch
 import gradio as gr
 import os
 import requests
@@ -43,8 +50,7 @@ def generate_radiology_description(
             length_penalty=1.0,
             num_return_sequences=1,
             conv_mode=conv_mode,
-            max_new_tokens=max_new_tokens,
-            device="cpu"
+            max_new_tokens=max_new_tokens
         )
         return output
     except Exception as e: