from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
import gradio as gr

model_id = "deepseek-ai/DeepSeek-Coder-V2-Instruct"

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)

generator = pipeline("text-generation", model=model, tokenizer=tokenizer)

def answer(prompt):
    outputs = generator(prompt, max_new_tokens=256, do_sample=False)
    return outputs[0]["generated_text"]

gr.Interface(fn=answer, inputs="text", outputs="text", title="DIAGMASTER IA").launch()