wizardlm_api / app.py
DR-Rakshitha's picture
Update app.py
62858fa
raw
history blame
574 Bytes
from gpt4all import GPT4All
import gradio as gr
# Specify the local path to the downloaded model file
model_path = "wizardlm-13b-v1.1-superhot-8k.ggmlv3.q4_0.bin"
# Initialize the model using the local path
model = GPT4All(model_path)
def generate_text(prompt):
result = model.generate(prompt)
return result
text_generation_interface = gr.Interface(
fn=generate_text,
inputs=[
gr.inputs.Textbox(label="Input Text"),
],
outputs=gr.outputs.Textbox(label="Generated Text"),
title="GPT-4 Text Generation",
).launch()
# model_name = ""