Upload app.py

app.py

@@ -3,6 +3,9 @@ import requests
 import time
 
 
+trial = 0
+
+
 API_URL = "https://cm7kxsqi3sekfih7.us-east-1.aws.endpoints.huggingface.cloud"
 headers = {
 	"Accept" : "application/json",
@@ -19,11 +22,20 @@ def query(payload):
 
 
 def run_model(text):
+	global trial
 	output = query({
 	"inputs": text,
 	"parameters": {}
 })
-	return output[0]['generated_text'] if output is not None else None
+	if output:
+		trial = 0
+		return output[0]['generated_text']
+	else:
+		trial += 1
+		if trial%2==1:
+			return 'Model is loading, please try again...'
+		else:
+			return 'Try one more time, it should work now...'
 	
 
 
@@ -45,10 +57,7 @@ examples = [
 
 def mode_run(text):
 	result = run_model(text)
-	if result:
-		return result
-	else:
-		return "Model is running please try again..."
+	return result
 
 
 demo = gr.Interface(fn=mode_run,