broadfield-dev commited on
Commit
d73df74
·
verified ·
1 Parent(s): 5855739

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,12 +1,14 @@
1
  import base64
2
  from flask import Flask, render_template, request, jsonify, Response, stream_with_context
 
3
  from vismem.core import SectorMemory
4
  from ai_engine import LocalModelHandler
 
5
  import json
6
  import os
7
-
8
  app = Flask(__name__)
9
 
 
10
  repo_id = os.getenv("LOCAL_MODEL_PATH", "google/gemma-2-9b-it")
11
  local_bot = LocalModelHandler(repo_id=repo_id)
12
 
 
1
  import base64
2
  from flask import Flask, render_template, request, jsonify, Response, stream_with_context
3
+ from huggingface_hub import login
4
  from vismem.core import SectorMemory
5
  from ai_engine import LocalModelHandler
6
+
7
  import json
8
  import os
 
9
  app = Flask(__name__)
10
 
11
+ if os.getenv("HF_TOKEN"): login(token=os.getenv("HF_TOKEN"))
12
  repo_id = os.getenv("LOCAL_MODEL_PATH", "google/gemma-2-9b-it")
13
  local_bot = LocalModelHandler(repo_id=repo_id)
14