File size: 408 Bytes
c7ca763
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from gradio_client import Client
from util.access_token import load_token


TOKEN = load_token("TOKEN_READ")

#Example Bot
def bot_response(input_text: str) -> str:
    client = Client("PlanBTech/Medflow-Bot", hf_token=TOKEN)
    result = client.predict(
        message=input_text,
        max_tokens=512,
        temperature=0.7,
        top_p=0.95,
        api_name="/chat",
    )

    return str(result)