Tonic commited on
Commit
1bfd63b
·
unverified ·
1 Parent(s): aa6bf8a
Files changed (1) hide show
  1. utils/oneclick.py +1 -1
utils/oneclick.py CHANGED
@@ -11,7 +11,7 @@ logger = logging.getLogger(__name__)
11
  HF_TOKEN = os.getenv("HF_TOKEN")
12
  if not HF_TOKEN:
13
  raise ValueError("HF_TOKEN environment variable not set.")
14
- client = InferenceClient(api_key=HF_TOKEN)client = InferenceClient() # Initialize the xAI client (adjust initialization as needed)
15
  MODEL_NAME = "meta-llama/Llama-3.3-70B-Instruct"
16
 
17
  def generate_ai_discharge_summary(patient_dict: Dict[str, str]) -> Optional[str]:
 
11
  HF_TOKEN = os.getenv("HF_TOKEN")
12
  if not HF_TOKEN:
13
  raise ValueError("HF_TOKEN environment variable not set.")
14
+ client = InferenceClient(api_key=HF_TOKEN)
15
  MODEL_NAME = "meta-llama/Llama-3.3-70B-Instruct"
16
 
17
  def generate_ai_discharge_summary(patient_dict: Dict[str, str]) -> Optional[str]: