Changing model and adding active wait

app.py

@@ -4,13 +4,16 @@ import time
 import requests
 import streamlit as st
 
-API_URL = "https://api-inference.huggingface.co/models/andrek/nb2nn"
+API_URL = "https://api-inference.huggingface.co/models/pere/nb-nn-translation"
 
 
 def translate(text, wait=True):
     headers = {"Authorization": f"Bearer {os.environ['BEARER']}"}
     payload  = {
         "inputs": text,
+        "options": {
+            "wait_for_model": not wait
+        }
     }
     response = requests.post(API_URL, headers=headers, json=payload)
     json_response = response.json()