Update app.py

app.py

@@ -2,6 +2,25 @@ import os
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
+# versions = {
+#     'v1': {
+#         'name': "milyiyo/paraphraser-german-mt5-small",
+#         'tokenizer': None,
+#         'model': None
+#     },
+#     'v2': {
+#         'name':"milyiyo/paraphraser-german-mt5-smal-v2",
+#         'tokenizer': None,
+#         'model': None
+#     },
+# }
+
+# versions['v1']['tokenizer'] = AutoTokenizer.from_pretrained(versions['v1']['name'])
+# versions['v1']['model'] = AutoModelForSeq2SeqLM.from_pretrained(versions['v1']['name'])
+
+# versions['v2']['tokenizer'] = AutoTokenizer.from_pretrained(versions['v2']['name'])
+# versions['v2']['model'] = AutoModelForSeq2SeqLM.from_pretrained(versions['v2']['name'])
+
 tokenizer = AutoTokenizer.from_pretrained("milyiyo/paraphraser-german-mt5-small")
 model = AutoModelForSeq2SeqLM.from_pretrained("milyiyo/paraphraser-german-mt5-small")