ccl-playground / app.py
darshil3011's picture
Update app.py
4376882
raw
history blame
675 Bytes
import gradio as gr
from gradio_client import Client
access_token = "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"
# Function to call the Gradio API
def call_gradio_api(user_input):
client = Client("https://traversaal-internal-rag-faiss-gpt3-5.hf.space/", hf_token=access_token)
result = client.predict(user_input, api_name="/predict")
return result[0], result[1], result[2]
# Interface for the Gradio app
iface = gr.Interface(
fn=call_gradio_api,
inputs="text",
outputs=["text","text","text"]
layout="vertical",
title="CCL Playground",
description="Enter a query to get response using RAG"
)
# Launch the Gradio app
iface.launch()