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
|