Update app.py

app.py

@@ -7,7 +7,7 @@ from sklearn.feature_extraction.text import TfidfVectorizer
 import pickle
 
 vectorizer = pickle.load(open("tfidf.pickle", "rb"))
-clf = pickle.load(open("classifier.pickle", "rb"))
+# clf = pickle.load(open("classifier.pickle", "rb"))
 
 example_context = "ফলস্বরূপ, ১৯৭৯ সালে, সনি এবং ফিলিপস একটি নতুন ডিজিটাল অডিও ডিস্ক ডিজাইন করার জন্য প্রকৌশলীদের একটি যৌথ টাস্ক ফোর্স গঠন করে। ইঞ্জিনিয়ার কিস শুহামার ইমমিনক এবং তোশিতাদা দোই এর নেতৃত্বে, গবেষণাটি লেজার এবং অপটিক্যাল ডিস্ক প্রযুক্তিকে এগিয়ে নিয়ে যায়। এক বছর পরীক্ষা-নিরীক্ষা ও আলোচনার পর টাস্ক ফোর্স রেড বুক সিডি-ডিএ স্ট্যান্ডার্ড তৈরি করে। প্রথম প্রকাশিত হয় ১৯৮০ সালে। আইইসি কর্তৃক ১৯৮৭ সালে আন্তর্জাতিক মান হিসেবে আনুষ্ঠানিকভাবে এই মান গৃহীত হয় এবং ১৯৯৬ সালে বিভিন্ন সংশোধনী মানের অংশ হয়ে ওঠে।'"
 example_answer = "১৯৮০"
@@ -46,14 +46,15 @@ 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: ',' '))
   
-  question = " ".join(text) 
-  correctness_pred = clf.predict(vectorizer.transform([question]))[0]
-  if correctness_pred == 1:
-    correctness = "Correct"
-  else : 
-    correctness = "Incorrect"
+  #question = " ".join(text) 
+  #correctness_pred = clf.predict(vectorizer.transform([question]))[0]
+  #if correctness_pred == 1:
+  #  correctness = "Correct"
+  #else : 
+  #  correctness = "Incorrect"
   
-  return question, correctness 
+  #return question, correctness 
+   return question 
   
   
 demo = gr.Interface(fn=generate_questions, inputs=[gr.Dropdown(label="Model", choices=["mt5-small","mt5-base","banglat5"],value="banglat5"),
@@ -69,8 +70,7 @@ 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'), 
-                                                             gr.Textbox(label='Grammatical Correctness')],
+                                                    outputs=[gr.Textbox(label='Question')],
                                                     examples=[["banglat5",example_context,example_answer]],
                                                     cache_examples=False,
                                                     title="Bangla Question Generation")