| #!/usr/bin/env python | |
| # coding: utf-8 | |
| # ## Using Gradio to create a simple interface. | |
| # | |
| # Check out the library on [github](https://github.com/gradio-app/gradio-UI) and see the [getting started](https://gradio.app/getting_started.html) page for more demos. | |
| # We'll start with a basic function that greets an input name. | |
| # In[1]: | |
| # get_ipython().system('pip install -q gradio') | |
| # Now we'll wrap this function with a Gradio interface. | |
| # In[2]: | |
| from transformers import pipeline | |
| import pandas as pd | |
| tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq") | |
| # In[ ]: | |
| tsqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-sqa") | |
| # In[ ]: | |
| mstqa = pipeline(task="table-question-answering", model="microsoft/tapex-large-finetuned-wikisql") | |
| # In[ ]: | |
| mswtqa = pipeline(task="table-question-answering", model="microsoft/tapex-large-finetuned-wtq") | |
| # In[6]: | |
| # table2 = pd.read_excel("/content/Sample.xlsx").astype(str) | |
| # table3 = table2.head(20) | |
| # In[7]: | |
| # table3 | |
| # In[ ]: | |
| #t4 = table3.reset_index() | |
| # table4 | |
| # In[9]: | |
| query = "what is the highest delta onu rx power?" | |
| query2 = "what is the lowest delta onu rx power?" | |
| query3 = "what is the most frequent login id?" | |
| query4 = "how many rows with nan values are there?" | |
| query5 = "how many S2 values are there" | |
| # In[11]: | |
| # result = tsqa(table=table3, query=query5)["answer"] | |
| # result | |
| # In[13]: | |
| #mstqa(table=table4, query=query1)["answer"] | |
| # In[14]: | |
| # mswtqa(table=table3, query=query5)["answer"] | |
| # In[15]: | |
| def main(filepath, query): | |
| table5 = pd.read_excel(filepath).head(20).astype(str) | |
| result = tsqa(table=table5, query=query)["answer"] | |
| return result | |
| #greet("World") | |
| # In[16]: | |
| import gradio as gr | |
| iface = gr.Interface( | |
| fn=main, | |
| inputs=[ | |
| gr.File(type="filepath", label="Upload XLSX file"), | |
| gr.Textbox(type="text", label="Enter text"), | |
| ], | |
| outputs=[gr.Textbox(type="text", label="Text Input Output")], | |
| title="TM TableQA Test", | |
| description="Upload an XLSX file and/or enter text, and the processed output will be displayed.", | |
| ) | |
| # Launch the Gradio interface | |
| iface.launch() | |
| # In[34]: | |
| import os | |
| import subprocess | |
| # Use subprocess to execute the shell command | |
| # subprocess.run(["jupyter", "nbconvert", "--to", "script", "--format", "script", "--output", "/content/", "/content/drive/MyDrive/Colab Notebooks/NEW TableQA-GRADIO: Hello World.ipynb"]) | |
| # In[19]: | |
| # get_ipython().system('gradio deploy') | |
| # That's all! Go ahead and open that share link in a new tab. Check out our [getting started](https://gradio.app/getting_started.html) page for more complicated demos. | |