Tonic commited on
Commit
22428d2
·
unverified ·
1 Parent(s): be39fa4

refactor matching algo

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,10 +1,12 @@
1
  # app.py
 
2
  from flask import Flask, render_template, request, send_file, redirect, url_for
3
  import os
4
  import logging
5
  from utils.meldrx import MeldRxAPI
6
  from utils.oneclick import generate_discharge_paper_one_click
7
- from xai import Grok # Adjust import based on actual xAI SDK
 
8
 
9
  logging.basicConfig(level=logging.DEBUG)
10
  logger = logging.getLogger(__name__)
@@ -24,7 +26,11 @@ meldrx_api = MeldRxAPI(
24
  redirect_uri=REDIRECT_URI
25
  )
26
 
27
- grok_client = Grok() # Adjust initialization as needed
 
 
 
 
28
 
29
  @app.route('/')
30
  def index():
 
1
  # app.py
2
+
3
  from flask import Flask, render_template, request, send_file, redirect, url_for
4
  import os
5
  import logging
6
  from utils.meldrx import MeldRxAPI
7
  from utils.oneclick import generate_discharge_paper_one_click
8
+ from huggingface_hub import InferenceClient
9
+
10
 
11
  logging.basicConfig(level=logging.DEBUG)
12
  logger = logging.getLogger(__name__)
 
26
  redirect_uri=REDIRECT_URI
27
  )
28
 
29
+ HF_TOKEN = os.getenv("HF_TOKEN")
30
+ if not HF_TOKEN:
31
+ raise ValueError("HF_TOKEN environment variable not set.")
32
+ client = InferenceClient(api_key=HF_TOKEN)
33
+ MODEL_NAME = "meta-llama/Llama-3.3-70B-Instruct"
34
 
35
  @app.route('/')
36
  def index():