response err handl

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 import requests
 import json
 import os
-API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-neo-2.7B"  # EleutherAI/gpt-neo-1.3B
+#API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-neo-2.7B"
+API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-neo-1.3B"
 apikey=os.environ.get('api_key')
 headers = {"Authorization": f"Bearer {apikey}"}
 def query(input_sentence,num,start):
@@ -12,7 +13,11 @@ def query(input_sentence,num,start):
        full_input=intial+"original:"+input_sentence + "\n paraphrase:"+start
        data = json.dumps({"inputs":full_input,"parameters":{"max_length":len(full_input.split())+70,"min_length":len(full_input.split())+70},"temperature":0.650+0.05*i})
        response = requests.request("POST", API_URL, headers=headers, data=data)
-       output=json.loads(response.content.decode("utf-8"))[0]['generated_text']
+       # https://stackoverflow.com/questions/16511337/correct-way-to-try-except-using-python-requests-module
+       if response.status_code != 200:
+          output = f'{response.status_code} problem, response={response.content.decode("utf-8")}'
+       else:
+          output=json.loads(response.content.decode("utf-8"))[0]['generated_text']
        paraphrase=output.split('paraphrase:',3)[-1]
        paraphrase_text=paraphrase.split('original:',1)[0]
        paraphrase_final.append( paraphrase_text.split('.',1)[0]+".")