Update app.py

app.py

@@ -40,10 +40,10 @@ def pre_query(recipient, recipient_name, sender, context, dates, input, model_id
         "parameters":{
                       "wait_for_model": True}
      }, API_URL)
-    if type(data) is dict:
-      return data['error']
-    else:
-      return data[0]['generated_text']
+    #if type(data) is dict:
+    #  return data['error']
+    #else:
+    return data[0]['generated_text']
   
   if model_id == "bigscience/bloom":
     input_string = "Write a professional email to my " + recipient.lower() + " starting with Hello " + recipient_name + ", about the subject " + context + " and the email should be based on this draft: " + input + ": Hello " + recipient_name + ",\n\n"
@@ -53,10 +53,10 @@ def pre_query(recipient, recipient_name, sender, context, dates, input, model_id
                       "return_full_text": False,
                       "wait_for_model": True}
      }, API_URL)
-    if type(data) is dict:
-      return data['error']
-    else:
-      return "Hello " + recipient_name + ",\n\n" + data[0]['generated_text'].replace(input_string,'')
+    #if type(data) is dict:
+    #  return data['error']
+    #else:
+    return "Hello " + recipient_name + ",\n\n" + data[0]['generated_text'].replace(input_string,'')
   
   if model_id == "GPT-3":
     return openai_query(recipient, 250, recipient_name, context, input)