Spaces:
Sleeping
Sleeping
import streamlit as st | |
from llama_cpp import Llama | |
llm = Llama.from_pretrained( | |
repo_id="Mykes/med_gemma7b_gguf", | |
filename="*Q4_K_M.gguf", | |
verbose=False | |
) | |
basic_prompt = "Below is the context which is your conversation history and the last user question. Write a response according the context and question. ### Context: user: Ответь мне на вопрос о моем здоровье. assistant: Конечно! Какой у Вас вопрос? ### Question: {question} ### Response:" | |
def generate_response(question): | |
model_input = basic_prompt.format(question=input_text) | |
if question: | |
output = llm( | |
model_input, # Prompt | |
max_tokens=32, # Generate up to 32 tokens, set to None to generate up to the end of the context window | |
stop=["<end_of_turn>"], | |
echo=False # Echo the prompt back in the output | |
) # Generate a completion, can also call create_completion | |
st.write(output["choices"][0]["text"]) | |
else: | |
st.write("Please enter a question to get a response.") | |
input_text = st.text_input('Задайте мне медицинский вопрос...') | |
# Button to trigger response generation | |
if st.button('Generate Response'): | |
generate_response(input_text) | |