Update app.py
Browse files
app.py
CHANGED
@@ -3,10 +3,12 @@ import streamlit as st
|
|
3 |
import subprocess
|
4 |
|
5 |
def run_llama(input):
|
6 |
-
output = subprocess.check_output(['./main', '-m', 'qunt4_0.bin', '-p', "Usuario: "+ input + ". Asistente:"])
|
|
|
7 |
response = output_str.split("Asistente:")[-1].strip()
|
8 |
return response
|
9 |
|
|
|
10 |
st.title("Llama Model")
|
11 |
|
12 |
input_text = st.text_input("Input Text", "")
|
|
|
3 |
import subprocess
|
4 |
|
5 |
def run_llama(input):
|
6 |
+
output = subprocess.check_output(['./main', '-m', 'qunt4_0.bin', '-p', "Usuario: "+ input + ". Asistente:"])
|
7 |
+
output_str = str(output.decode('utf-8'))
|
8 |
response = output_str.split("Asistente:")[-1].strip()
|
9 |
return response
|
10 |
|
11 |
+
|
12 |
st.title("Llama Model")
|
13 |
|
14 |
input_text = st.text_input("Input Text", "")
|