generationGH / app.py
TanyaVM's picture
Update app.py
e3b880a verified
Raw
History Blame Contribute Delete
899 Bytes
import gradio as gr
import os,json
import requests
from huggingface_hub import InferenceClient
# Your endpoint URL
HF_ENDPOINT = "https://fkm4akb80r7t3p6v.us-east4.gcp.endpoints.huggingface.cloud"
HF_API_TOKEN = os.getenv('API_TKN_ALL') # Replace with your actual HF token
headers = {
"Authorization": f"Bearer {HF_API_TOKEN}",
"Content-Type": "application/json",
"Accept" : "application/json"
}
def generate_text(user_text):
user_text = {"inputs": "User write a Gherkin scenario for user login and save it."}
response = requests.post(HF_ENDPOINT, headers=headers, json=user_text)
return response.json()
# Gradio Interface
gr.Interface(
fn=generate_text,
inputs=gr.Textbox(lines=4, label="Enter your task"),
outputs=gr.Textbox(lines=6, label="Generated Text"),
title="Text Generation via Hugging Face Inference Endpoint"
).launch()