GENAI / app.py
Vikas1994's picture
Updated generalization for question
1e0c25e
raw
history blame contribute delete
861 Bytes
import streamlit as st
import os
from langchain import PromptTemplate, HuggingFaceHub, LLMChain
x = st.slider('Select a value')
st.write(x, 'squared is', x * x)
# https://cobusgreyling.medium.com/langchain-creating-large-language-model-llm-applications-via-huggingface-192423883a74
# !pip install langchain[all]
template = """Question: {question}"""
HUGGING_FACE_API_KEY= os.environ.get("HUGGING_FACE_API_KEY")
Answer= "Let's think step by step."
prompt = PromptTemplate(template=template, input_variables=["question"])
llm=HuggingFaceHub(repo_id="Writer/palmyra-small", model_kwargs={"temperature":1e-10},huggingfacehub_api_token=HUGGING_FACE_API_KEY)
question=st.text_input("Input the question",'')
# question = "When was Google founded?"
# print(llm.run(question))
if question:
d=LLMChain(llm=llm,prompt=prompt)
st.write("out",d.run(question))