Update app.py

app.py

@@ -1,6 +1,10 @@
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import gradio as gr
+import pickle
+
+vectorizer = pickle.load(open("tfidf.pickle", "rb"))
+clf = pickle.load(open("classifier.pickle", "rb"))
 
 example_context = "ফলস্বরূপ, ১৯৭৯ সালে, সনি এবং ফিলিপস একটি নতুন ডিজিটাল অডিও ডিস্ক ডিজাইন করার জন্য প্রকৌশলীদের একটি যৌথ টাস্ক ফোর্স গঠন করে। ইঞ্জিনিয়ার কিস শুহামার ইমমিনক এবং তোশিতাদা দোই এর নেতৃত্বে, গবেষণাটি লেজার এবং অপটিক্যাল ডিস্ক প্রযুক্তিকে এগিয়ে নিয়ে যায়। এক বছর পরীক্ষা-নিরীক্ষা ও আলোচনার পর টাস্ক ফোর্স রেড বুক সিডি-ডিএ স্ট্যান্ডার্ড তৈরি করে। প্রথম প্রকাশিত হয় ১৯৮০ সালে। আইইসি কর্তৃক ১৯৮৭ সালে আন্তর্জাতিক মান হিসেবে আনুষ্ঠানিকভাবে এই মান গৃহীত হয় এবং ১৯৯৬ সালে বিভিন্ন সংশোধনী মানের অংশ হয়ে ওঠে।'"
 example_answer = "১৯৮০"
@@ -39,7 +43,14 @@ def generate_questions(model_choice,context,answer,numReturnSequences=1,num_beam
   for id in generated_ids:
     text.append(tokenizer.decode(id,skip_special_tokens=True,clean_up_tokenization_spaces=True).replace('question: ',' '))
   
-  return " ".join(text)
+  question = " ".join(text) 
+  correctness_pred = clf.predict(vectorizer.transform([question]))[0]
+  if correctness_pred == 1:
+    correctness = "Correct"
+  else : 
+    correctness = "Incorrect"
+  
+  return question, correctness 
   
   
 demo = gr.Interface(fn=generate_questions, inputs=[gr.Dropdown(label="Model", choices=["mt5-small","mt5-base","banglat5"],value="banglat5"),
@@ -55,7 +66,8 @@ demo = gr.Interface(fn=generate_questions, inputs=[gr.Dropdown(label="Model", ch
                                                     gr.Slider(0, 1, value=None, label="Top P/Nucleus Sampling"),
                                                     gr.Slider(0, 1, value=None, label="Temperature") ] ,
                                                     # output
-                                                    outputs=gr.Textbox(label='Question'),
+                                                    outputs=[gr.Textbox(label='Question'), 
+                                                             gr.Textbox(label='Grammatical Correctness')],
                                                     examples=[["banglat5",example_context,example_answer]],
                                                     cache_examples=False,
                                                     title="Bangla Question Generation")