Update app.py

app.py

@@ -4,7 +4,6 @@ import gradio as gr
 import os
 
 import sys
-# sys.path.append('/app')
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 
 from ONNX0630 import main as predict_smiles
@@ -16,28 +15,11 @@ import logging
 # 设置日志
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
-
-# Gradio interface
-# def gradio_predict(image):
-#     try:
-#         # Save the uploaded image
-#         temp_path = "temp_image.png"
-#         image.save(temp_path)
-        
-#         # Call the model function
-#         smiles = predict_smiles(temp_path)
-        
-#         # Clean up
-#         os.remove(temp_path)
-        
-#         return smiles
-#     except Exception as e:
-#         return f"Error: {str(e)}"
-    
+   
 def gradio_predict(image):
     try:
         # 使用临时文件目录保存图片
-        with tempfile.NamedTemporaryFile(suffix=".png", dir="/home/user/app", delete=False) as temp_file:
+        with tempfile.NamedTemporaryFile(suffix=".png", dir="/temp", delete=False) as temp_file:
             temp_path = temp_file.name
             logger.info(f"temp_path: {temp_path}")
             image.save(temp_path)