Dhruv's picture
WIP: gradio frontend code.
74e4c5d
import gradio as gr
from gradio import TabbedInterface
from gh_api_info import GithubRepoAccess
from transformers import AutoModelForCausalLM, AutoTokenizer
import requests
from dotenv import dotenv_values
def getRepoInformation(input_queries):
config = dotenv_values(".env")
obj = GithubRepoAccess(config["GITHUB_TOKEN"])
repo = obj.getRepo(input_queries)
return repo
def ressurectingFrontend():
search = gr.Interface(
fn= getRepoInformation,
inputs= gr.inputs.Textbox(lines=1, label="Search"),
outputs= gr.outputs.Textbox(label="Output"),
title="Search Github",
description="build and ressurect old github inputs",
article="follow the link to github repo and our submission in bioML",
)
return search.launch()
ressurectingFrontend()