redditbot / app.py
Tonic's picture
Update app.py
472a4e3
raw
history blame
592 Bytes
import os
import gradio as gr
import gradio_client as grc
# Access environment variables
hf_token = os.environ.get("HF_TOKEN")
inference_endpoint = os.environ.get("INFERENCE_ENDPOINT")
# Check if the environment variables are set
if hf_token is None or inference_endpoint is None:
raise ValueError("HF_TOKEN or INFERENCE_ENDPOINT environment variable not set")
secrets = {"HF_TOKEN": hf_token, "INFERENCE_ENDPOINT": inference_endpoint}
client = grc.Client.duplicate("TeamTonic/coherebot", private=False, secrets=secrets, sleep_timeout=2880)
client.deploy_discord(api_names=["predict"])