PeteBleackley
Fine tuned model was producing nonsense, try mt5
3b5865c
raw
history blame
341 Bytes
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Sun Aug 11 10:11:53 2024
@author: peter
"""
import gradio as gr
import tokenizers
import transformers
pipeline = transformers.pipeline('translation_en_to_ha',
model='google/mt5-small')
demo = gr.Interface.from_pipeline(pipeline)
demo.launch()