Update Space

app.py

@@ -6,12 +6,8 @@ API_URL = "https://api-inference.huggingface.co/models/gowtham58/T_TL"
 headers = {"Authorization": f"Bearer {TK}"}
 
 def get_output(text):
-	response = requests.post(API_URL, headers=headers, json={"inputs": text,})
+	response = requests.post(API_URL, headers=headers, json={"inputs": text,"wait_for_model":True})
 	response = response.json()
-	#print(response)
-	while response[0]=='error':
-		response = requests.post(API_URL, headers=headers, json={"inputs": text,})
-		response = response.json()
 	
 	return response[0]['generated_text']