nikunjcepatel's picture
Create app.py
66ff3e9 verified
raw
history blame
772 Bytes
import gradio as gr
import requests
def chat_with_openrouter(prompt):
url = "https://api.openrouter.ai/v1/chat/completions"
headers = {"Authorization": "Bearer OpenRounter_API_KEY"}
data = {
"model": "openai/gpt-4o-mini-2024-07-18",
"messages": [{"role": "user", "content": prompt}],
"top_p:" 1,
"temperature:" 1,
"repetition_penalty:" 1,
"transforms": []
}
response = requests.post(url, headers=headers, json=data)
return response.json().get("choices", [{}])[0].get("message", {}).get("content", "No response")
iface = gr.Interface(
fn=chat_with_openrouter,
inputs=gr.Textbox(lines=2, placeholder="Ask me anything"),
outputs="text",
title="Chat with OpenRouter",
)
iface.launch()