Update main.py
Browse files
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
|