| # -*- coding: utf-8 -*- | |
| import streamlit as st | |
| import subprocess | |
| def run_llama(input): | |
| output = subprocess.check_output(['./main', '-m', 'ggml-vicuna-7b-1.1-q4_0.bin', '-p', "User: "+ input + ". Assistant:"]) | |
| output_str = str(output.decode('utf-8')) | |
| response = output_str.split("Asistente:")[-1].strip() | |
| return response | |
| st.title("Llama Model") | |
| input_text = st.text_input("Input Text", "") | |
| if st.button("Run"): | |
| output = run_llama(input_text) | |
| st.write(output) |