Spaces:
Runtime error
Runtime error
from fastapi import FastAPI, Request | |
from fastapi.middleware.cors import CORSMiddleware | |
import asyncio | |
from Linlada import Chatbot, ConversationStyle | |
app = FastAPI() | |
app.add_middleware( | |
CORSMiddleware, | |
allow_origins=["*"], | |
allow_methods=["*"], | |
allow_headers=["*"], | |
allow_credentials=True, | |
) | |
async def generate(prompt): | |
bot = await Chatbot.create() | |
result = await bot.ask(prompt=prompt, conversation_style=ConversationStyle.precise) | |
return result | |
def read_root(): | |
return "Hello, I'm Linlada" | |
def hi(hello: str): | |
return {"text": hello} | |
async def generate_image_route(request: Request): | |
data = await request.json() | |
prompt = data['prompt'] | |
result = await generate(prompt) | |
return result | |
def generate_image_route(prompt: str): | |
loop = asyncio.new_event_loop() | |
asyncio.set_event_loop(loop) | |
result = loop.run_until_complete(generate(prompt)) | |
loop.close() | |
return result |