Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ for gpu in gpus:
|
|
16 |
try:
|
17 |
tf.config.experimental.set_memory_growth(gpu, True)
|
18 |
except RuntimeError as e:
|
19 |
-
print(
|
20 |
# Initialize AI Core for TB analysis
|
21 |
ai_core = AICoreAGIX()
|
22 |
|
@@ -39,14 +39,14 @@ async def diagnose_tb_async(image_file, audio_file):
|
|
39 |
pass
|
40 |
|
41 |
return (
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
)
|
51 |
|
52 |
def diagnose_tb(image_file, audio_file):
|
@@ -56,7 +56,7 @@ def upload_and_finetune(jsonl_file):
|
|
56 |
if jsonl_file is None:
|
57 |
return "Please upload a .jsonl file to fine-tune Codriao."
|
58 |
|
59 |
-
save_path =
|
60 |
os.makedirs("training_data", exist_ok=True)
|
61 |
|
62 |
with open(save_path, "wb") as f:
|
@@ -71,7 +71,7 @@ def upload_and_finetune(jsonl_file):
|
|
71 |
except:
|
72 |
pass
|
73 |
|
74 |
-
return "
|
75 |
|
76 |
def get_latest_model():
|
77 |
return "Download the latest fine-tuned Codriao model here: https://huggingface.co/Raiff1982/codriao-finetuned"
|
|
|
16 |
try:
|
17 |
tf.config.experimental.set_memory_growth(gpu, True)
|
18 |
except RuntimeError as e:
|
19 |
+
print("[TF] GPU memory growth config error: {e}")
|
20 |
# Initialize AI Core for TB analysis
|
21 |
ai_core = AICoreAGIX()
|
22 |
|
|
|
39 |
pass
|
40 |
|
41 |
return (
|
42 |
+
"**TB Risk Level:** {result['tb_risk']}\n\n"
|
43 |
+
"**Image Result:** {result['image_analysis']['result']} "
|
44 |
+
"(Confidence: {result['image_analysis']['confidence']:.2f})\n\n"
|
45 |
+
"**Audio Result:** {result['audio_analysis']['result']} "
|
46 |
+
"(Confidence: {result['audio_analysis']['confidence']:.2f})\n\n"
|
47 |
+
"**Ethical Analysis:** {result['ethical_analysis']}\n\n"
|
48 |
+
"**Explanation:** {result['explanation']}\n\n"
|
49 |
+
"**Shareable Link:** {result['shareable_link']}"
|
50 |
)
|
51 |
|
52 |
def diagnose_tb(image_file, audio_file):
|
|
|
56 |
if jsonl_file is None:
|
57 |
return "Please upload a .jsonl file to fine-tune Codriao."
|
58 |
|
59 |
+
save_path = "./training_data/{jsonl_file.name}"
|
60 |
os.makedirs("training_data", exist_ok=True)
|
61 |
|
62 |
with open(save_path, "wb") as f:
|
|
|
71 |
except:
|
72 |
pass
|
73 |
|
74 |
+
return "
Fine-tuning complete! Model updated and stored."
|
75 |
|
76 |
def get_latest_model():
|
77 |
return "Download the latest fine-tuned Codriao model here: https://huggingface.co/Raiff1982/codriao-finetuned"
|