Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import streamlit as st
|
|
3 |
import subprocess
|
4 |
|
5 |
def run_llama(input):
|
6 |
-
output = subprocess.check_output(['./main', '-m', '
|
7 |
output_str = str(output.decode('utf-8'))
|
8 |
response = output_str.split("Asistente:")[-1].strip()
|
9 |
return response
|
|
|
3 |
import subprocess
|
4 |
|
5 |
def run_llama(input):
|
6 |
+
output = subprocess.check_output(['./main', '-m', 'ggml-vicuna-7b-1.1-q4_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
|