Spaces:
Build error
Build error
from langchain import PromptTemplate, LLMChain | |
from huggingface_hub import hf_hub_download | |
from langchain.llms import GPT4All | |
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler | |
import os | |
template =""" | |
You are a friendly chatbot assistant you are going to evaluate a Github Repo i analysed all the repos and made a report | |
i'm gonna give a report you just refer the below table | |
Table: | |
score Risk | |
1 - 5 Low - Simple block | |
6 - 10 Low - Well structured and stable block | |
11 - 20 Moderate - Slightly complex block | |
21 - 30 More than moderate - More complex block | |
31 - 40 High - Complex block, alarming | |
41+ Very high - Error-prone, unstable block | |
based on the score and you have to return only one most complex repo which has more score from report | |
when the score is high thats the complex repo and | |
if all the repos are at the same score choose randomly one | |
Report: {question} | |
Answer:""" | |
prompt = PromptTemplate(template=template, input_variables=["question"]) | |
#hf_hub_download(repo_id="dnato/ggml-gpt4all-j-v1.3-groovy.bin", filename="ggml-gpt4all-j-v1.3-groovy.bin", local_dir="/code") | |
local_path= os.getcwd() + "/ggml-gpt4all-j-v1.3-groovy.bin" | |
llm = GPT4All(model=local_path,callbacks=[StreamingStdOutCallbackHandler()] ) | |
llm_chain = LLMChain(prompt=prompt, llm=llm) | |