Update modeling_ostlm.py

modeling_ostlm.py

@@ -105,27 +105,3 @@ class OSTLMModel(PreTrainedModel, GenerationMixin):
 AutoConfig.register("ostlm", OSTLMConfig)
 AutoModelForSeq2SeqLM.register(OSTLMConfig, OSTLMModel)
 
-def translate(text, model_path="./ostlm_v1_final"):
-    tokenizer = AutoTokenizer.from_pretrained(model_path)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
-    model.to("cuda" if torch.cuda.is_available() else "cpu").eval()
-
-    inputs = tokenizer(text, return_tensors="pt", padding=True).to(model.device)
-
-    with torch.no_grad():
-        outputs = model.generate(
-            input_ids=inputs["input_ids"],
-            max_length=4,
-            num_beams=3, # Beam search מצומצם יותר ליציבות
-            decoder_start_token_id=tokenizer.cls_token_id,
-            eos_token_id=tokenizer.sep_token_id,
-            pad_token_id=tokenizer.pad_token_id,
-        )
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-print("--- 🤖 OSTLM v1: Structure Fixed ---")
-try:
-    print(f"Result: {translate('Who?')}")
-except Exception as e:
-    import traceback
-    traceback.print_exc()