Update app.py

app.py

@@ -4,28 +4,31 @@ import json
 
 def predict_aphasia(json_input):
     try:
-        # 解析 JSON 輸入
+        # 解析JSON輸入
         data = json.loads(json_input)
         
-        # 預處理數據給你的模型
-        processed_data = preprocess(data)
+        # 載入你的模型
+        model = torch.load('pytorch_model.bin', map_location='cpu')
+        model.eval()
         
-        # 模型推理
+        # 處理數據並預測
         with torch.no_grad():
-            prediction = model(processed_data)
-            aphasia_type = get_aphasia_type(prediction)
+            # 這裡放你的資料前處理和模型推理代碼
+            result = model(processed_data)
+            predicted_type = result  # 例如 "BROCA", "WERNICKE" 等
         
-        return f"預測的失語症類型: {aphasia_type}"
+        return f"預測的失語症類型: {predicted_type}"
     
     except Exception as e:
         return f"錯誤: {str(e)}"
 
+# 建立界面
 demo = gr.Interface(
     fn=predict_aphasia,
     inputs=gr.Textbox(
         label="輸入對話數據 (JSON格式)", 
         lines=20,
-        placeholder="貼上你的JSON數據..."
+        placeholder="請貼上您的JSON數據..."
     ),
     outputs=gr.Textbox(label="分類結果"),
     title="失語症類型分類器",