Update app.py

app.py

@@ -130,41 +130,78 @@ def important_sentences(filepath, no_of_sentences=5):
   text=""
   for index,sent in enumerate(extractive_summary.split(".")):
     if sent!='':text+=str(index+1)+". "+str(sent).strip()+".\n\n"
-  return (gr.Textbox.update(text),gr.Button.update(visible=False),gr.Textbox.update(visible=False))
+  return (gr.Textbox.update(text),gr.Button.update(visible=False),gr.Textbox.update(visible=False),gr.Dropdown.update(visible=False))
 
 def summarize(filepath):
   extractedInformation=read(filepath)
   extractedInformation=' '.join(extractedInformation.split('\n'))
   output = summarizer(extractedInformation, max_length=int(len(extractedInformation)/6), min_length=int(len(extractedInformation)/10), do_sample=False)
-  return (gr.Textbox.update(output[0]["summary_text"]),gr.Button.update(visible=False),gr.Textbox.update(visible=False))
+  return (gr.Textbox.update(output[0]["summary_text"]),gr.Button.update(visible=False),gr.Textbox.update(visible=False),gr.Dropdown.update(visible=False))
 
-def Question_Answer(filepath,question):
+def Question_Answer(filepath,question,mod):
   extractedInformation=read(filepath)
   extractedInformation=' '.join(extractedInformation.split('\n'))
-  question_answerer = pipeline("question-answering", model="SMD00/QA_model-roberta")
+  if mod=="Roberta":
+    question_answerer = pipeline("question-answering", model="SMD00/QA_model-roberta")
+  else :
+    question_answerer = pipeline("question-answering", model="SMD00/QA_model-distilbert")
   obj=question_answerer(question=question, context=extractedInformation)
   return obj['answer']
 
-
 def show_fn():
     return (gr.Textbox.update(visible=True),gr.Button.update(visible=True),gr.Textbox.update(""))
 
+def show_fn():
+    return (gr.Textbox.update(visible=True),gr.Button.update(visible=True),gr.Dropdown.update(visible=True),gr.Textbox.update(""))
+def dummy_fn(x):
+  return x
+
 with gr.Blocks() as demo:
     gr.Markdown("# **PicSum**")
     gr.Markdown("Gradio demo for PicSum project. You can give an image as input and select any of the three buttons. It generates summary, important sentences and answers questions related to context.")
     img=gr.components.Image(type="filepath", label="Input Image")
-    
+
     with gr.Row():
         summary = gr.Button(value="Summary")
         sentence = gr.Button(value="Important Sentences")
         quesAndAns = gr.Button(value="Question and Answers")
 
-    ques_box = gr.Textbox(label="Question",interactive=True,visible=False)
+    mode=gr.Dropdown(["Roberta","DistilBert"],label="Model",info="Choose a model",visible=False)
+    ques_box = gr.Textbox(label="Question",info="Enter a Question",interactive=True,visible=False)
     submit= gr.Button(value="Submit",visible=False)
     out=gr.Textbox(label="Generated Text")
-    summary.click(fn=summarize,inputs=[img],outputs=[out,submit,ques_box])
-    sentence.click(fn=important_sentences,inputs=[img],outputs=[out,submit,ques_box])
-    quesAndAns.click(fn=show_fn,outputs=[submit,ques_box,out])
-    submit.click(fn=Question_Answer,inputs=[img,ques_box],outputs=[out])
-
+    summary.click(fn=summarize,inputs=[img],outputs=[out,submit,ques_box,mode])
+    sentence.click(fn=important_sentences,inputs=[img],outputs=[out,submit,ques_box,mode])
+    quesAndAns.click(fn=show_fn,outputs=[submit,ques_box,mode,out])
+    submit.click(fn=Question_Answer,inputs=[img,ques_box,mode],outputs=[out])
+    gr.Markdown("## Image Examples")
+    with gr.Row():
+        gr.Examples(
+            examples=[ "a.png"],
+            inputs=img,
+            outputs=img,
+            fn=dummy_fn,
+            cache_examples=True,
+        )
+        gr.Examples(
+            examples=[ "b.png"],
+            inputs=img,
+            outputs=img,
+            fn=dummy_fn,
+            cache_examples=True,
+        )
+        gr.Examples(
+            examples=[ "c.png"],
+            inputs=img,
+            outputs=img,
+            fn=dummy_fn,
+            cache_examples=True,
+        )
+                gr.Examples(
+            examples=[ "d.png"],
+            inputs=img,
+            outputs=img,
+            fn=dummy_fn,
+            cache_examples=True,
+        )
 demo.launch(debug=True)