upd gr ver

README.md

@@ -4,7 +4,7 @@ emoji: 💬
 colorFrom: blue
 colorTo: gray
 sdk: gradio
-sdk_version: 5.22.0
+sdk_version: 6.3.0
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -23,4 +23,4 @@ if __name__ == "__main__":
         with gr.Tab(_L("真实 DeepSeek R1 Qwen 7B 模型")):
             DeepSeek_R1_Qwen_7B()
 
-    demo.launch()
+    demo.launch(css="#gradio-share-link-button-0 { display: none; }")

modules/deepseek.py

@@ -1,6 +1,4 @@
 import torch
-import modelscope
-import huggingface_hub
 import gradio as gr
 from threading import Thread
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
@@ -33,11 +31,18 @@ DESCRIPTION = (
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 if device == torch.device("cuda"):
-    MODEL_DIR = (
-        huggingface_hub.snapshot_download(MODEL_ID, cache_dir="./__pycache__")
-        if EN_US
-        else modelscope.snapshot_download(MODEL_ID, cache_dir="./__pycache__")
-    )
+    if EN_US:
+        import modelscope
+
+        MODEL_DIR = modelscope.snapshot_download(MODEL_ID, cache_dir="./__pycache__")
+
+    else:
+        import huggingface_hub
+
+        MODEL_DIR = huggingface_hub.snapshot_download(
+            MODEL_ID, cache_dir="./__pycache__"
+        )
+
     tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
     model = AutoModelForCausalLM.from_pretrained(MODEL_DIR, device_map="auto")
 

requirements.txt

@@ -1,5 +1,3 @@
-torch==2.6.0+cu118
--f https://download.pytorch.org/whl/torch
-openai
 accelerate
-modelscope[framework]==1.24.0
+openai
+torch