code

app.py

@@ -1,12 +1,18 @@
 import spaces
 import gradio as gr
-import transformers
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import os
 
+if not os.path.exists("./phi-4"):
+    os.system('huggingface-cli download matteogeniaccio/phi-4 --local-dir ./phi-4 --include "phi-4/*"')
+
+import transformers
+
 # 初始化pipeline
 pipeline = transformers.pipeline(
     "text-generation",
-    model="microsoft/phi-4",
+    model="./phi-4/phi-4",
     model_kwargs={"torch_dtype": "auto"},
     device_map="auto",
 )