Update app.py

app.py

@@ -22,11 +22,20 @@ from pydantic import BaseModel
 from typing import Annotated
 from transformers import BertTokenizerFast, EncoderDecoderModel
 import torch
+import threading
+import random
+import string
+import time
 
 from fastapi import Form
 
 class Query(BaseModel):
     text: str
+
+
+class Query2(BaseModel):
+    text: str
+    host:str
    
 # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # tokenizer = BertTokenizerFast.from_pretrained('mrm8488/bert-small2bert-small-finetuned-cnn_daily_mail-summarization')
@@ -103,4 +112,41 @@ async def get_answer(q: Query ):
       
    
     return "hello"
-    
+
+
+@app.post("/large")
+async def get_answer(q: Query2 ):
+
+    N = 20
+    res = ''.join(random.choices(string.ascii_uppercase +
+                             string.digits, k=N))
+    res= res+ str(time.time())
+
+    id= res
+    text = q.text
+    host= q.host
+    t = threading.Thread(target=do_ML, args=(id,text,host))  
+    t.start()
+
+
+    
+    return JSONResponse({"id":id})
+      
+   
+    return "hello"
+
+    
+import requests 
+    
+def do_ML(id:str,long_text:str,host:str):
+    try:
+        
+        r= generate_summary(long_text)
+        data={"id":id,"result":r}
+        x=requests.post(host,data= data)
+        print(x.text)
+        
+
+    except:
+        print("Error  occured id= "+id)
+