Update app.py

app.py

@@ -101,15 +101,6 @@ def main(urls):
 @spaces.GPU()
 def classify_website(url):
     global model, tokenizer  # Declare model and tokenizer as global variables
-
-    urls = [url]
-    results_shop = main(urls)
-
-    # Convert results to DataFrame
-    df_result_train_more = pd.DataFrame(results_shop)
-    text = df_result_train_more['text'][0]
-    translated = GoogleTranslator(source='auto', target='en').translate(text[:4990])
-
     
     try:
         # Load the model and tokenizer if they are not already loaded
@@ -125,6 +116,16 @@ def classify_website(url):
             )
             FastLanguageModel.for_inference(model)  # Enable native 2x faster inference
 
+
+        urls = [url]
+        results_shop = main(urls)
+    
+        # Convert results to DataFrame
+        df_result_train_more = pd.DataFrame(results_shop)
+        text = df_result_train_more['text'][0]
+        translated = GoogleTranslator(source='auto', target='en').translate(text[:4990])
+    
+            
         prompt = f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
 
 ### Instruction: