deepseekphilo1 / app.py
methodya's picture
Update app.py
a53a69e verified
raw
history blame
445 Bytes
import gradio as gr
from transformers import pipeline
# استخدام النموذج مباشرة بدون LoRA
pipe = pipeline(
"text-generation",
model="deepseek-ai/deepseek-coder-1.3b-base",
device_map="auto"
)
def generate_response(prompt):
result = pipe(prompt, max_length=100)
return result[0]['generated_text']
iface = gr.Interface(
fn=generate_response,
inputs="text",
outputs="text"
)
iface.launch()