Jofthomas HF staff commited on
Commit
bb41ed4
·
1 Parent(s): 0bde9db

Update TextGen/router.py

Browse files
Files changed (1) hide show
  1. TextGen/router.py +2 -2
TextGen/router.py CHANGED
@@ -314,8 +314,8 @@ def generate_voice_eleven(request: Request, message: VoiceMessage = None):
314
  # return StreamingResponse(audio_stream(),media_type="audio/mpeg")
315
  @app.post("/generate_song")
316
  @app.get("/generate_song")
317
- async def generate_song(request:SongRequest,request: Request):
318
- if request.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
319
  return
320
  backstory=main_npc_system_prompts["Bard"]
321
  print(backstory)
 
314
  # return StreamingResponse(audio_stream(),media_type="audio/mpeg")
315
  @app.post("/generate_song")
316
  @app.get("/generate_song")
317
+ async def generate_song(request:SongRequest,httprequest: Request):
318
+ if httprequest.headers.get("origin") != "https://jofthomas-everchanging-quest.static.hf.space":
319
  return
320
  backstory=main_npc_system_prompts["Bard"]
321
  print(backstory)