Update app.py

app.py

@@ -86,18 +86,15 @@ def sumy(payload):
 
 
 def query(payload):
-    API_URL = "https://api-inference.huggingface.co/models/gpt2"
+    API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
     data = json.dumps(payload)
     response = requests.request("POST", API_URL, headers=headers, data=data)
     return json.loads(response.content.decode("utf-8"))
 
 def google(name): 
-            if "give" in name:
-              r_c = name.split("give")[1]
+            if "give" in name or "reason" in name or "result" in name or "step" in name:
 
-              result_count = int(r_c.split()[0])
-              #print(result_count)
-              name = name.split("give")[0]
+              result_count = 2
               print(name)
 
             else:
@@ -142,18 +139,18 @@ def google(name):
                  f_result = sumy({"inputs":temp,"parameters": {"do_sample": False,"max_length":300}})
                  return f_result[0]['summary_text']
             else:
-              for r in text.split("\n\n"): 
+              n=1
+              for r in text.split("\n\n")[2:-2]:
+                if len(r)>10:
                   if "..." in r:
                     r = r.split("...")
-                    print(r[0].strip(),1)
-                    w = query(r[0].strip())[0]['generated_text'].strip()
-                    if "\n\n" in w:
-                      print("=")
-                    f_result = f_result +w+"\n\n"
+                    w = query(r[0].replace("\xa0",""))
+                    f_result = f_result + str(n)+"\t"+(w[0]['summary_text'])+"\n\n"+r"\\"
                   else:
-                    print("=")
-                    f_result = f_result + r.strip()+"\n\n"
-              
+                    #print(r[:-1])
+                    w = query(r[:-1])
+                    f_result = f_result + str(n)+"\t"+(w[0]['summary_text'])+"\n\n"+r"\\"
+                  n=n+1
               return f_result
 
 def greet(name1):