Medapp / config.py
mgbam's picture
Update config.py
5121009 verified
raw
history blame
423 Bytes
import os
# Secrets and environment variables
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
MY_PUBMED_EMAIL = os.getenv("MY_PUBMED_EMAIL")
HF_TOKEN = os.getenv("HF_TOKEN")
# Default OpenAI Model
OPENAI_DEFAULT_MODEL = "gpt-4o"
# Hugging Face Model for Medical Image Analysis
HF_IMAGE_MODEL = "nlpconnect/vit-gpt2-image-captioning"
# PubMed and summarization settings
MAX_PUBMED_RESULTS = 5
SUMMARIZATION_CHUNK_SIZE = 512