deploy

app/main.py

@@ -51,15 +51,16 @@ async def email_input_root(request: Request):
 
 @app.post("/subject_generation/")
 async def create_upload_files(request: Request , paragraphInput: str = Form(...)):
-    global email_inp
     print("hi im in subject gen", paragraphInput )
-    print("########################",request)
-    result = subject_gen.subject_gen_func( paragraphInput)
-    print(result)
-    result = result.split("<subject>")[1].replace("<eos>","")
+    result = subject_gen.subject_gen_func(paragraphInput)
+    # print(result)
+    if(len(result.split("<subject>"))>1):
+        subject = result.split("<subject>")[1].replace("<eos>","")
+    else :
+        subject = "Please Try Again"
     return templates.TemplateResponse("subject_generation.html", {"request": request,
-                                                       "result":result,
-                                                       "email": paragraphInput,})
+                                                       "result":subject,
+                                                       "email":paragraphInput,})
 
 
 # Set all CORS enabled origins

app/pyfiles/subject_gen.py

@@ -25,4 +25,4 @@ def subject_gen_func(email):
 
 
     generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
-    return generated_text.split("<subject>")[1].replace("<eos>","")
+    return generated_text