File size: 554 Bytes
ec97e44 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | from resume import CVParse
from github import relevance_filter
from transformers import AutoTokenizer, pipeline, AutoModelForCausalLM
import torch
def init():
tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/deepseek-coder-1.3b-instruct")
model = AutoModelForCausalLM.from_pretrained("deepseek-ai/deepseek-coder-1.3b-instruct")
device = "cuda" if torch.cuda.is_available() else "cpu"
gen_pipeline = pipeline(task="text-generation", model=model, tokenizer = tokenizer)
def run(CV, Repo_link):
return "hello world" |