Create app.py

app.py

@@ -1,3 +1,6 @@
+# Install required packages
+!pip install huggingface_hub
+
 import gradio as gr
 import torch
 import torch.nn as nn
@@ -139,15 +142,37 @@ class UltimateCommunicationAnalyzer:
             # Load custom intent model
             self.intent_model = MultiLabelIntentClassifier("distilbert-base-uncased", 6)
             
-            # Try to load local model first, then from HF if available
+            # Try to load from HuggingFace first, then local file
             try:
-                checkpoint = torch.load('intent_detection_model.pth', map_location='cpu')
-                self.intent_model.load_state_dict(checkpoint['model_state_dict'])
-                logger.info("✅ Intent detection model loaded from local file!")
-            except FileNotFoundError:
-                logger.warning("Local intent model not found, using fallback...")
-                # Could load from HF here if uploaded
-                raise Exception("Intent model not found - please ensure intent_detection_model.pth exists")
+                logger.info("Attempting to load from HuggingFace: SamanthaStorm/intentanalyzer...")
+                # Download the pytorch_model.bin from HuggingFace
+                from huggingface_hub import hf_hub_download
+                
+                model_path = hf_hub_download(
+                    repo_id="SamanthaStorm/intentanalyzer",
+                    filename="pytorch_model.bin",
+                    cache_dir="./models"
+                )
+                
+                # Load the state dict
+                state_dict = torch.load(model_path, map_location='cpu')
+                self.intent_model.load_state_dict(state_dict)
+                logger.info("✅ Intent detection model loaded from HuggingFace!")
+                
+            except Exception as hf_error:
+                logger.warning(f"HuggingFace download failed: {hf_error}")
+                logger.info("Trying local file...")
+                
+                # Fallback to local file
+                try:
+                    checkpoint = torch.load('intent_detection_model.pth', map_location='cpu')
+                    self.intent_model.load_state_dict(checkpoint['model_state_dict'])
+                    logger.info("✅ Intent detection model loaded from local file!")
+                except FileNotFoundError:
+                    logger.error("❌ Neither HuggingFace nor local model found!")
+                    raise Exception("Intent model not found. Please either:")
+                    print("1. Ensure 'intent_detection_model.pth' exists locally, OR")
+                    print("2. Make sure SamanthaStorm/intentanalyzer is properly uploaded to HuggingFace")
                 
         except Exception as e:
             logger.error(f"❌ Error loading intent model: {e}")
@@ -553,7 +578,8 @@ if __name__ == "__main__":
     except Exception as e:
         logger.error(f"❌ Failed to launch app: {e}")
         print(f"Error: {e}")
-        print("\nMake sure both model files are available:")
-        print("1. Fallacy model: Available from HuggingFace (SamanthaStorm/fallacyfinder)")
-        print("2. Intent model: Local file 'intent_detection_model.pth' required")
+        print("\nMake sure both models are available:")
+        print("1. Fallacy model: SamanthaStorm/fallacyfinder (auto-downloaded)")
+        print("2. Intent model: SamanthaStorm/intentanalyzer (auto-downloaded)")
+        print("3. Or ensure 'intent_detection_model.pth' exists locally")
         raise