Spaces:
Sleeping
Sleeping
File size: 2,967 Bytes
a2cccdb e7f4e46 a2cccdb d9c6906 a2cccdb d9c6906 a2cccdb d9c6906 a2cccdb e7f4e46 a2cccdb 9bf72c1 d9c6906 a2cccdb d9c6906 9bf72c1 d9c6906 9bf72c1 d9c6906 9bf72c1 d9c6906 9bf72c1 d9c6906 9bf72c1 a2cccdb |
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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
import streamlit as st
from llmware.prompts import Prompt
import requests
import io
import PyPDF2
def register_gguf_model():
prompter = Prompt()
your_model_name = "llama"
hf_repo_name = "TheBloke/Llama-2-7B-Chat-GGUF"
model_file = "llama-2-7b-chat.Q5_K_S.gguf"
print("registering models")
prompter.model_catalog.register_gguf_model(your_model_name,hf_repo_name, model_file, prompt_wrapper="open_chat")
your_model_name = "open_gpt4"
hf_repo_name = "TheBloke/Open_Gpt4_8x7B-GGUF"
model_file = "open_gpt4_8x7b.Q4_K_M.gguf"
prompter.model_catalog.register_gguf_model(your_model_name,hf_repo_name, model_file, prompt_wrapper="open_chat")
your_model_name = "phi2"
hf_repo_name = "TheBloke/phi-2-GGUF"
model_file = "phi-2.Q4_K_M.gguf"
prompter.model_catalog.register_gguf_model(your_model_name,hf_repo_name, model_file, prompt_wrapper="open_chat")
your_model_name = "mistral"
hf_repo_name = "TheBloke/Mistral-7B-Instruct-v0.2-GGUF"
model_file = "mistral-7b-instruct-v0.2.Q4_K_M.gguf"
prompter.model_catalog.register_gguf_model(your_model_name,hf_repo_name, model_file, prompt_wrapper="open_chat")
# print("loading model")
# prompter.load_model(your_model_name)
return prompter
def load_pdf_from_url(url):
r = requests.get(url)
f = io.BytesIO(r.content)
file = PyPDF2.PdfReader(f)
return file
def load_pdf_content(pdf):
content = ""
for page in pdf.pages:
content += page.extract_text()
return content
def main():
st.title("BetterZila RAG Enabled LLM")
with st.spinner("Registering Models for use..."):
prompter = register_gguf_model()
model_name = st.selectbox("Select Model", ["llama", "open_gpt4", "phi2", "mistral"])
st.write("You selected: ", model_name)
with st.spinner("Loading model..."):
prompter.load_model(model_name)
st.success("Model loaded!")
with st.spinner("Loading PDF content from the assignment URL..."):
url = "https://pgcag.files.wordpress.com/2010/01/48lawsofpower.pdf"
pdf = load_pdf_from_url(url)
content = load_pdf_content(pdf)
print("Loaded PDF content")
st.success("PDF content loaded!")
queries = ['Can you give me an example from history where the enemy was crushed totally from the book?', "What's the point of making myself less accessible?", "Can you tell me the story of Queen Elizabeth I from this 48 laws of power book?"]
for query in queries:
st.subheader(f"Query: {query}")
with st.spinner("Generating response..."):
print("Query: ", query)
resp = prompter.prompt_main(query, context=content)
response = resp['llm_response']
print("Time taken: ", response['usage']['processing_time'])
print("Response: ", response)
st.success("Response generated!")
st.write(response)
if __name__ == "__main__":
main()
|