Spaces:
Sleeping
Sleeping
File size: 909 Bytes
66ff3e9 8658e51 b944409 66ff3e9 8658e51 c307a1d 8658e51 c307a1d 8658e51 66ff3e9 8658e51 66ff3e9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import gradio as gr
import requests
import json
import os
# Retrieve the Open Router API Key from the Space secrets
API_KEY = os.getenv("OpenRounter_API_KEY")
def chat_with_openrouter(input_text):
response = requests.post(
url="https://openrouter.ai/api/v1/chat/completions",
headers={
"Authorization": f"Bearer {API_KEY}"
},
data=json.dumps({
"model": "openai/gpt-4o-mini-2024-07-18", # Optional
"messages": [
{"role": "user", "content": input_text}
]
"top_p": 1,
"temperature": 1,
"frequency_penalty": 0,
"presence_penalty": 0,
"repetition_penalty": 1,
"top_k": 0,
})
)
return response.json().get("choices", [{}])[0].get("message", {}).get("content", "No response")
# Create Gradio interface
iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
iface.launch()
|