Update app.py

app.py

@@ -1,9 +1,9 @@
+from all_models import models
 from flask import Flask, request, jsonify, send_file
-from flask_cors import CORS  # Import CORS
-from externalmod import gr_Interface_load  # Import the required function for loading models
+from flask_cors import CORS
+from gradio_client import Client
+from externalmod import gr_Interface_load
 import os
-from all_models import models
-
 
 app = Flask(__name__)
 CORS(app)  # Enable CORS for all routes
@@ -19,9 +19,7 @@ def load_model(model_name):
         print(f"Error loading model: {error}")
         return None
 
-# Load your model here
-#model_name = "Yntec/Lexicon"  # Change this to your actual model name
-#model_interface = load_model(model_name)
+
 
 @app.route('/predict', methods=['POST'])
 def predict():