File size: 826 Bytes
f71c233 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import json
import openai
from ai_scientist.perform_review import load_paper, perform_review
client = openai.OpenAI(api_key="ollama", base_url="http://localhost:11434/v1")
model = "mistral-nemo"
# Load paper from pdf file (raw text)
# paper_txt = load_paper("review_test.pdf")
### include the path to the aider chat of the research, from results/{MODEL-USED}
paper_txt = load_paper("/home/ubuntu/ollama-ai-sci/AI-Scientist/results/nanoGPT/20240918_182054_refined_augmented_preprocessing/20240918_182054_refined_augmented_preprocessing_aider.txt")
# Get the review dict of the review
review = perform_review(
paper_txt,
model,
client,
num_reflections=5,
num_fs_examples=1,
num_reviews_ensemble=1,
temperature=0.1,
)
with open("review.txt", "w") as f:
f.write(json.dumps(review, indent=4))
|