Update app.py

app.py

@@ -3,8 +3,8 @@ import gradio as gr
 
 # Import pandas datasets, transformers, torch
 import pandas as pd
-import torch
-import tensorflow as tf
+#import torch
+#import tensorflow as tf
 from datasets import load_dataset
 
 from transformers import (
@@ -38,8 +38,8 @@ processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-larg
 model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
 
 #Load transformer for translating captions from English to Spanish
-trans_tokenizer = AutoTokenizer.from_pretrained("arthur-neto/english-to-spanish-lang-translation-model")
-trans_model = AutoModelForSeq2SeqLM.from_pretrained("arthur-neto/english-to-spanish-lang-translation-model", from_tf=True)
+trans_tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-tc-big-en-es")
+trans_model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-tc-big-en-es")