anuragmalik
commited on
Commit
•
ce6c237
1
Parent(s):
92cdb3b
Update app.py
Browse files
app.py
CHANGED
@@ -1,16 +1,17 @@
|
|
1 |
# Q&A Chatbot
|
2 |
-
from langchain.llms import OpenAI
|
3 |
from langchain import HuggingFaceHub
|
4 |
-
|
5 |
|
6 |
-
|
7 |
|
8 |
import streamlit as st
|
9 |
import os
|
10 |
|
|
|
11 |
|
12 |
## Function to load OpenAI model and get respones
|
13 |
-
os.environ["HUGGINGFACEHUB_API_TOKEN"]=
|
14 |
llm_huggingface=HuggingFaceHub(repo_id="google/flan-t5-xxl",model_kwargs={"temperature":0.6,"max_length":64})
|
15 |
|
16 |
|
|
|
1 |
# Q&A Chatbot
|
2 |
+
#from langchain.llms import OpenAI
|
3 |
from langchain import HuggingFaceHub
|
4 |
+
from dotenv import load_dotenv
|
5 |
|
6 |
+
load_dotenv() # take environment variables from .env.
|
7 |
|
8 |
import streamlit as st
|
9 |
import os
|
10 |
|
11 |
+
huggingface_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
|
12 |
|
13 |
## Function to load OpenAI model and get respones
|
14 |
+
os.environ["HUGGINGFACEHUB_API_TOKEN"]=huggingface_token
|
15 |
llm_huggingface=HuggingFaceHub(repo_id="google/flan-t5-xxl",model_kwargs={"temperature":0.6,"max_length":64})
|
16 |
|
17 |
|