Spaces:
Runtime error
Runtime error
| from models import llm, retrieval | |
| import gradio as gr | |
| def run_query(question, db_path="data/videos.db", | |
| num_rel_segments=5, | |
| llm_model="gpt-3.5-turbo-0125", | |
| llm_temp=0.1): | |
| relevant_segments = retrieval.get_relevant_segments(question, | |
| db_path=db_path, | |
| n_results=num_rel_segments) | |
| answer = llm.answer_with_context(question, | |
| relevant_segments, | |
| model=llm_model, | |
| temperature=llm_temp) | |
| return answer | |
| if __name__ == "__main__": | |
| demo = gr.Interface(fn=run_query, | |
| inputs="text", | |
| outputs="text", | |
| title="Fitness Q&A", | |
| description="Ask me a question about fitness! If I can, I'll provide a link to one of Dr. Andrew Huberman's videos with more information.", | |
| theme="soft", | |
| examples=["How can I promote muscle recovery?", "How does caffeine affect my workout?"] | |
| ) | |
| demo.launch(share=True) | |