MitchellKil commited on
Commit
674ea65
·
verified ·
1 Parent(s): 42c6620

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,6 +3,7 @@ from pydantic import BaseModel
3
  import torch
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
  from datetime import datetime
 
6
 
7
  # my fine tuned model
8
  MODEL_NAME = "MitchellKil/gaelic-ipa-byt5"
@@ -46,7 +47,7 @@ def text_to_ipa(text: str) -> str:
46
 
47
  class Feedback(BaseModel):
48
  type: str
49
- feedback: Optional[str]
50
  input_text: str
51
  translation: str
52
  phonetic: str
 
3
  import torch
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
  from datetime import datetime
6
+ import json
7
 
8
  # my fine tuned model
9
  MODEL_NAME = "MitchellKil/gaelic-ipa-byt5"
 
47
 
48
  class Feedback(BaseModel):
49
  type: str
50
+ feedback: str
51
  input_text: str
52
  translation: str
53
  phonetic: str