Gurgen-Blbulyan
adding files for app
08cc25a
raw
history blame
426 Bytes
import gradio as gr
from inference import Inference
encoder_model_name='google/vit-large-patch32-224-in21k'
decoder_model_name='gpt2-large'
inference = Inference(
decoder_model_name=decoder_model_name,
)
def generate_text(video):
generated_text = inference.generate_text(video, encoder_model_name)
return generated_text
app = gr.Interface(fn=generate_text, inputs='video', outputs='text')
app.launch(share=True)