Update app.py

app.py

@@ -11,18 +11,18 @@ from huggingface_hub import hf_hub_download
 api = HfApi()
 app = FastAPI() 
 
-MODEL_NAME = "MitchellKil/gaelic-ipa-byt5" # my fine tuned model
+modelName = "MitchellKil/gaelic-ipa-byt5" # my fine tuned model
 DATASET_REPO = "mitchellkil/gaelic-app-feedback" # file to write feedback to
 
-HF_TOKEN = os.environ["HF_TOKEN"] # set in Space secrets
+HF_TOKEN = os.environ["HF_TOKEN"] # set in secrets
 
 
-print("Loading Gaelic → IPA model from HF")
+print("Loading IPA model")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)
+tokenizer = AutoTokenizer.from_pretrained(modelName)
+model = AutoModelForSeq2SeqLM.from_pretrained(modelName).to(device)
 model.eval()
 
 print("Model loaded") 
@@ -33,13 +33,13 @@ class TextRequest(BaseModel):
 
 def text_to_ipa(text: str) -> str:
     # Input text - > target text for trained model
-    prompt = f"convert this from Gaelic to IPA: {text}"
+    prompt = f"convert Gaelic to IPA: {text}"
 
     inputs = tokenizer(
         prompt,
         return_tensors="pt",
         truncation=True,
-        max_length=128
+        max_length=110
     ).to(device)
 
     with torch.no_grad():
@@ -83,10 +83,9 @@ def save_feedback(data: Feedback):
         )
         with open(local_file, "r", encoding="utf-8") as f:
             lines = f.readlines()
-    except:
-        lines = []
+    
 
-    lines.append(json.dumps(entry) + "\n") # Append
+    lines.append(json.dumps(entry) + "\n") # append
 
     temp_path = "feedback_temp.jsonl" # create temp
     with open(temp_path, "w", encoding="utf-8") as f:
@@ -98,8 +97,8 @@ def save_feedback(data: Feedback):
         repo_id=DATASET_REPO,
         repo_type="dataset",
         token=HF_TOKEN,
-        commit_message="Append new feedback"
+        commit_message="New feedback added"
     )
 
-    return {"status": "Thank you for your feedback"}
+    return {"status": "Thanks for your feedback"}