Spaces:
Runtime error
Runtime error
import gradio as gr | |
import chromadb | |
import pandas as pd | |
def greet(name): | |
client = chromadb.Client() | |
collection = client.create_collection("bolivian-recipes") | |
df = pd.read_parquet("hf://datasets/asoria/bolivian-recipes@~parquet/default/other/0000.parquet") | |
text_column = "preparation" | |
ids = [str(i) for i in range(df.shape[0])] | |
documents = df[text_column].to_list() | |
metadatas = df.drop(text_column, axis=1).to_dict("records") | |
collection.add(ids=ids, documents=documents, metadatas=metadatas) | |
results = collection.query(query_texts=["sausage"], n_results=4) | |
print(results) | |
return "Hello " + name + "!!" | |
iface = gr.Interface(fn=greet, inputs="text", outputs="text") | |
iface.launch() |