Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,8 @@
|
|
1 |
import gradio as gr
|
2 |
-
from inference import evo_chat_predict, get_model_config, get_gpt_response
|
3 |
import pandas as pd
|
4 |
import csv
|
|
|
5 |
from datetime import datetime
|
6 |
|
7 |
feedback_log = []
|
@@ -55,10 +56,23 @@ with gr.Blocks(theme=gr.themes.Base(), css="body { background-color: #0f0f0f; co
|
|
55 |
|
56 |
return evo_text, gpt_text, stats_text, updated_df
|
57 |
|
58 |
-
# 🔁
|
59 |
def retrain_evo():
|
60 |
-
|
61 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
62 |
|
63 |
# 🧹 Clear UI
|
64 |
def clear_fields():
|
|
|
1 |
import gradio as gr
|
2 |
+
from inference import evo_chat_predict, get_model_config, get_gpt_response
|
3 |
import pandas as pd
|
4 |
import csv
|
5 |
+
import os
|
6 |
from datetime import datetime
|
7 |
|
8 |
feedback_log = []
|
|
|
56 |
|
57 |
return evo_text, gpt_text, stats_text, updated_df
|
58 |
|
59 |
+
# 🔁 Retrain Evo using retrain.py
|
60 |
def retrain_evo():
|
61 |
+
if not feedback_log:
|
62 |
+
return "⚠️ No feedback data to retrain from."
|
63 |
+
|
64 |
+
# Save log to feedback_log.csv in required format
|
65 |
+
with open("feedback_log.csv", "w", newline="") as f:
|
66 |
+
writer = csv.writer(f)
|
67 |
+
writer.writerow(["prompt", "context", "label"])
|
68 |
+
for row in feedback_log:
|
69 |
+
prompt = row[0]
|
70 |
+
context = row[6] # context_used
|
71 |
+
label = 0 if row[3] == row[1] else 1 # 0 if answer == option1 else 1
|
72 |
+
writer.writerow([prompt, context, label])
|
73 |
+
|
74 |
+
os.system("python retrain.py")
|
75 |
+
return "✅ Evo retrained from feedback_log.csv."
|
76 |
|
77 |
# 🧹 Clear UI
|
78 |
def clear_fields():
|