File size: 779 Bytes
6a38c0c 4e26dbe 4f29f66 2672a33 4e26dbe | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | from langchain import HuggingFaceHub
from langchain.llms import OpenAI
from dotenv import load_dotenv
import streamlit as st
import os
load_dotenv()
def get_openai_response(question):
# try:
# llm = OpenAI(model="text-davinci-003",openai_api_key=os.getenv("OPEN_API_KEY"),temperature=0.6)
# response=llm(question)
# except:
llm_hugg = HuggingFaceHub(repo_id="google/flan-t5-large",model_kwargs={'temperature':0, "max_length":64})
response=llm_hugg(question)
return response
st.set_page_config(page_title="Q&A Demo")
st.header("Langchain Aplication")
input = st.text_input("Input: ",key="input")
response = get_openai_response(input)
submit = st.button("Ask Question")
if submit:
st.subheader("The Response is:")
st.write(response)
|