Spaces:
Runtime error
Runtime error
| import zipfile | |
| import gradio as gr | |
| import tensorflow as tf | |
| from transformers import pipeline | |
| import tensorflow_text as text | |
| pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-tr-en") | |
| reloaded_model = tf.saved_model.load('./') | |
| def main(input): | |
| pure_out = reloaded_model.predict(pipe(input)) | |
| out = tf.argmax(pure_out) | |
| labels = ['Visual', 'Kinesthetic', 'Auditory'] | |
| outS = [] | |
| index = out.index(1) | |
| label = labels[index] | |
| outS.append(label) | |
| return outS[0] | |
| iface = gr.Interface(fn=main, inputs="text", outputs="text") | |
| iface.launch() |