TRaw's picture
Update app.py
30b907b verified
raw
history blame
468 Bytes
import gradio as gr
import requests
import os
API_URL = "https://api-inference.huggingface.co/models/voxta/ehartford_dolphin-2_2-yi-34b-3bpw-exl2"
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.json()
def query_wrapper(text):
payload = {"inputs": text}
output = query(payload)
return output
iface = gr.Interface(
fn=query_wrapper,
inputs="text",
outputs="json"
)
iface.launch()