Spaces:
Runtime error
Runtime error
File size: 1,498 Bytes
0a533a6 2bd81e9 cd93ac8 0a533a6 cd93ac8 0a533a6 cd93ac8 0a533a6 244d820 c79e14b 0a533a6 c79e14b 0a533a6 c79e14b 1e31df6 0a533a6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
import streamlit as st
import json
from gradio_client import Client
# Load data.json
with open('data.json') as f:
data = json.load(f)
client_endpoint = "https://olivier-truong-mistral-super-fast.hf.space/"
def ask_question(question):
if question in data:
return data[question]
else:
client = Client(client_endpoint)
result = client.predict(
question,
0.9, # Temperature
2000, # Max new tokens
0.9, # Top-p (nucleus sampling)
1.2, # Repetition penalty
api_name="/chat"
)
answer = result["data"]
data[question] = answer
with open('data.json', 'w') as f:
json.dump(data, f)
return answer
def typewriter(text: str, speed: int):
tokens = text.split()
container = st.empty()
for index in range(len(tokens) + 1):
curr_full_text = " ".join(tokens[:index])
container.markdown(curr_full_text)
time.sleep(1 / speed)
def main():
st.title("Question Answering System")
question = st.text_input("Ask your question:")
if st.button("Ask"):
if question:
answer_placeholder = st.empty()
answer_placeholder.write("Thinking...")
answer = ask_question(question)
answer_placeholder.empty()
typewriter("Answer: " + answer, speed=50)
else:
st.write("Please enter a question.")
if __name__ == "__main__":
main()
|