Hugging Face's logo

Spaces:
unaexoo
/
DiseaseRAG
Runtime error

App Files Community
unaexoo commited on
Commit
150b862
·
verified ·
1 Parent(s): 94c7bd0

Update disease.py

Browse files
Files changed (1) hide show
  1. disease.py +2 -2
disease.py CHANGED
@@ -57,8 +57,8 @@ vectorstore = Chroma.from_documents(documents=split_documents, embedding=embeddi
57
  retriever = vectorstore.as_retriever()
58
 
59
  # %%
60
- # 단계 6: Perplexity AI R1-1776 모델 로드
61
- model_name = "perplexity-ai/r1-1776"
62
 
63
  tokenizer = AutoTokenizer.from_pretrained(model_name)
64
  model = AutoModelForCausalLM.from_pretrained(model_name)
 
57
  retriever = vectorstore.as_retriever()
58
 
59
  # %%
60
+ # 단계 6: 모델 로드
61
+ model_name = "qihoo360/TinyR1-32B-Preview"
62
 
63
  tokenizer = AutoTokenizer.from_pretrained(model_name)
64
  model = AutoModelForCausalLM.from_pretrained(model_name)