Spaces:

UcsTurkey
/

fine-tune-inference-test

Paused

ciyidogan commited on May 9

Commit

cc493ed

verified ·

1 Parent(s): 53de39f

Update fine_tune_inference_test.py

Files changed (1) hide show

fine_tune_inference_test.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import os
 import threading
 import uvicorn
-from fastapi import FastAPI, Request
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from datasets import load_dataset
@@ -22,9 +22,31 @@ chat_history = []
 class Message(BaseModel):
     user_input: str
-@app.get("/")
-def health():
-    return {"status": "ok"}
 @app.post("/chat")
 def chat(msg: Message):

 import os
 import threading
 import uvicorn
+from fastapi import FastAPI, Request, HTMLResponse
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from datasets import load_dataset
 class Message(BaseModel):
     user_input: str
+@app.get("/", response_class=HTMLResponse)
+def root():
+    return """
+    <html>
+    <head><title>Fine-Tune Chat</title></head>
+    <body>
+        <h2>📘 Fine-tune Chat Test</h2>
+        <textarea id="input" rows="4" cols="60" placeholder="Bir şeyler yaz..."></textarea><br><br>
+        <button onclick="send()">Gönder</button>
+        <pre id="output"></pre>
+        <script>
+            async function send() {
+                const input = document.getElementById("input").value;
+                const res = await fetch("/chat", {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ user_input: input })
+                });
+                const data = await res.json();
+                document.getElementById("output").innerText = data.answer || data.error || "Hata oluştu.";
+            }
+        </script>
+    </body>
+    </html>
+    """
 @app.post("/chat")
 def chat(msg: Message):