Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from huggingface_hub import hf_hub_download | |
| from llama_cpp import Llama | |
| hf_hub_download(repo_id="LLukas22/gpt4all-lora-quantized-ggjt", filename="ggjt-model.bin", local_dir=".") | |
| llm = Llama(model_path="./ggjt-model.bin") | |
| ins = '''### Instruction: | |
| {} | |
| ### Response: | |
| ''' | |
| theme = gr.themes.Monochrome( | |
| primary_hue="indigo", | |
| secondary_hue="blue", | |
| neutral_hue="slate", | |
| radius_size=gr.themes.sizes.radius_sm, | |
| font=[gr.themes.GoogleFont("Open Sans"), "ui-sans-serif", "system-ui", "sans-serif"], | |
| ) | |
| def generate(instruction): | |
| result = "" | |
| for x in llm(ins.format(instruction), stop=['### Instruction:', '### End'], stream=True): | |
| result += x['choices'][0]['text'] | |
| yield result | |
| gr.ChatInterface( | |
| fn=generate, | |
| chatbot=gr.Chatbot(height=300), | |
| textbox=gr.Textbox(placeholder="Ask me a question"), | |
| title="Healthcare Bot", | |
| description="Ask the Healthcare Bot any question", | |
| examples = [ | |
| "Give me treatments for heart disease", | |
| "I hate exercise, what else can I do to treat my high blood pressure", | |
| "How can I avoid lung disease", | |
| ], | |
| theme=theme, | |
| ).launch() | |