Spaces:

ruslanmv
/

DeepSeek-R1-Chatbot

Running

ruslanmv commited on 3 days ago

Commit

cec1f02

verified ·

1 Parent(s): d212f4f

Update models.py

Files changed (1) hide show

models.py CHANGED Viewed

@@ -1,5 +1,4 @@
-import gradio as gr
-# Load and return the model interface
 def load_model():
-    return gr.Interface.load("models/deepseek-ai/DeepSeek-R1")

+from transformers import pipeline
 def load_model():
+    return pipeline("text-generation", model="deepseek-ai/DeepSeek-R1")