cmpatino's picture
Handle installatio of custom .whl
a834a31
raw
history blame
1.15 kB
import os
import gradio as gr
import pkg_resources
from dotenv import load_dotenv
load_dotenv()
def package_installed(package_name):
try:
pkg_resources.get_distribution(package_name)
except pkg_resources.DistributionNotFound:
return False
else:
return True
def answer(query):
return agent.answer(query=query)
if not package_installed("cv_assistant"):
os.system("pip install cv_assistant-0.1-py2.py3-none-any.whl")
from cv_assistant.agent import Agent
agent = Agent(
faiss_index_path="./content_assets/docs.index",
faise_store_path="./content_assets/faiss_store.pkl",
)
description = """
### Ask about my experience, skills, and education!
I built this using [Gradio](https://gradio.app) and [LangChain](https://langchain.readthedocs.io/en/latest/).
"""
title = "Career Chatbot"
iface = gr.Interface(
fn=answer,
inputs=gr.Textbox(
value="What's his experience in recommender systems?", label="Question"
),
outputs=gr.Textbox(label="Answer"),
description=description,
title=title,
analytics_enabled=True,
allow_flagging="auto",
)
iface.launch()