aminaj commited on
Commit
7e178d1
·
verified ·
1 Parent(s): 545ef3d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -40,6 +40,7 @@ async def lifespan(app: FastAPI):
40
  ## load the Groq API key
41
  os.environ['GROQ_API_KEY'] = os.getenv("GROQ_API_KEY")
42
  os.environ['HF_TOKEN'] = os.getenv("HF_TOKEN")
 
43
  global image_to_text
44
  image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
45
  yield
 
40
  ## load the Groq API key
41
  os.environ['GROQ_API_KEY'] = os.getenv("GROQ_API_KEY")
42
  os.environ['HF_TOKEN'] = os.getenv("HF_TOKEN")
43
+ os.environ['NGROK_AUTHTOKEN'] = os.getenv("NGROK_AUTHTOKEN")
44
  global image_to_text
45
  image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
46
  yield