Update app.py

app.py

@@ -1,8 +1,8 @@
 import os
 import torch
+from flask import Flask, render_template, request, redirect, url_for
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from peft import PeftModel
-from flask import Flask, request, jsonify, render_template
 
 # --- Load Model & Tokenizer ---
 
@@ -35,23 +35,25 @@ print("Model ready!")
 
 app = Flask(__name__)
 
-@app.route("/")
+@app.route("/", methods=["GET"])
 def index():
     return render_template("index.html")
 
 @app.route("/generate", methods=["POST"])
 def generate():
-    data = request.get_json()
-    prompt = data.get("prompt")
+    prompt = request.form["prompt"]
 
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     outputs = model.generate(**inputs, max_new_tokens=100)
     text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    
-    return jsonify({
-        "generated_text": text
-    })
+
+    return redirect(url_for("result", generated_text=text))
+
+@app.route("/result")
+def result():
+    generated_text = request.args.get("generated_text", "")
+    return render_template("result.html", generated_text=generated_text)
 
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860))
-    app.run(host="0.0.0.0", port=port)
+    app.run(host="0.0.0.0", port=port)