Spaces:

ikraamkb
/

Summarization

Running

ikraamkb commited on 17 days ago

Commit

b7dc4fe

verified ·

1 Parent(s): 4a74e6a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -73,7 +73,7 @@ import torch
 from transformers import ViltProcessor, ViltForQuestionAnswering, AutoTokenizer, AutoModelForCausalLM
 from gtts import gTTS
 import gradio as gr
 app = FastAPI()
 # Load VQA Model

 from transformers import ViltProcessor, ViltForQuestionAnswering, AutoTokenizer, AutoModelForCausalLM
 from gtts import gTTS
 import gradio as gr
+from transformers import AutoModelForSeq2SeqLM
 app = FastAPI()
 # Load VQA Model