upgrade gradio version

README.md

@@ -4,7 +4,7 @@ emoji: 🏃
 colorFrom: indigo
 colorTo: gray
 sdk: gradio
-sdk_version: 4.2.0
+sdk_version: 5.1.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -1,21 +1,17 @@
-import gradio as gr
-import pandas as pd
 import torch
+import gradio as gr
 from googletrans import Translator
 from transformers import T5Tokenizer
 from transformers import T5ForConditionalGeneration
 from transformers import BartForConditionalGeneration
 from transformers import BartTokenizer
-from transformers import pipeline
+from transformers import PreTrainedModel
+from transformers import PreTrainedTokenizer
 
 tokenizer = BartTokenizer.from_pretrained('facebook/bart-large-cnn')
 model = BartForConditionalGeneration.from_pretrained('facebook/bart-large-cnn')
 
 
-
-from transformers import PreTrainedModel
-from transformers import PreTrainedTokenizer
-
 # Question launcher
 class E2EQGPipeline:
     def __init__(
@@ -24,7 +20,7 @@ class E2EQGPipeline:
         tokenizer: PreTrainedTokenizer
     ):
 
-        self.device = "cuda" if torch.cuda.is_available()  else "cpu"
+        self.device = "cuda" if torch.cuda.is_available() else "cpu"
 
         self.model = model
         self.tokenizer = tokenizer
@@ -100,9 +96,10 @@ def generate_summary(text):
 
 
 def process(text):
-    return generate_summary(text), generate_questions(text)
+    print(generate_questions(text))
+    return generate_summary(text)
 
 
 textbox = gr.Textbox(label="Pega el text aca:", placeholder="Texto...", lines=15)
-demo = gr.Interface(fn=process, inputs=textbox, outputs=["text", "text"])
+demo = gr.Interface(fn=process, inputs=textbox, outputs="text")
 demo.launch()

requirements.txt

@@ -1,5 +1,8 @@
-gradio==4.2.0
+gradio==5.1.0
 transformers
 torch
 accelerate
-googletrans-py
+# We are using this fork since the orignal google library uses an old package version of httpx
+# which is not compatible with the last version of gradio
+googletrans-py
+sentencepiece