fix port

app.py

@@ -6,7 +6,6 @@ import re
 
 MODEL_ID = "cahya/gpt2-small-indonesian-522M"
 
-# Load tokenizer + model on CPU
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 model = AutoModelForCausalLM.from_pretrained(MODEL_ID, torch_dtype=torch.float32, device_map={"": "cpu"})
 
@@ -26,21 +25,18 @@ def generate_ui(prompt):
     cleaned_text = clean_and_extract_json(raw_text)
     try:
         parsed = json.loads(cleaned_text)
-        # Check fields and strip spaces
         script = parsed.get("script", "").strip()
         caption = parsed.get("caption", "").strip()
         hashtags = parsed.get("hashtags", [])
         if not script or not caption or not isinstance(hashtags, list):
-            raise ValueError("Missing or invalid fields in JSON")
+            raise ValueError("Missing or invalid fields")
         return {
             "script": script,
             "caption": caption,
             "hashtags": hashtags
         }
     except Exception:
-        # Fallback: don't include the prompt, just first sentence or default text
         fallback_script = raw_text.strip()
-        # Remove prompt repetition if possible
         if prompt in fallback_script:
             fallback_script = fallback_script.replace(prompt, "").strip()
         fallback_caption = fallback_script.split(".")[0].strip() or "Simak info menarik ini!"