RehanKingggg commited on
Commit
ae3c917
·
verified ·
1 Parent(s): 5c44b83

Upload model_options.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. model_options.py +80 -0
model_options.py ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """
3
+ Model Options - Brello EI 0
4
+ Created by Epic Systems | Engineered by Rehan Temkar
5
+
6
+ Demonstrates different model options for Brello EI 0.
7
+ """
8
+
9
+ from brello_ei_0 import BrelloEI0
10
+
11
+ def test_model_option(model_path, description):
12
+ """Test a specific model option"""
13
+ print(f"\n🤖 Testing: {description}")
14
+ print(f"Model: {model_path}")
15
+ print("-" * 50)
16
+
17
+ try:
18
+ # Load the model
19
+ model = BrelloEI0(
20
+ model_path=model_path,
21
+ load_in_4bit=False
22
+ )
23
+
24
+ # Test emotional intelligence
25
+ test_message = "I'm feeling really stressed about my presentation tomorrow."
26
+ response = model.generate_response(test_message)
27
+
28
+ print(f"Input: {test_message}")
29
+ print(f"Response: {response}")
30
+ print("✅ Model working!")
31
+
32
+ return True
33
+ except Exception as e:
34
+ print(f"❌ Model failed: {e}")
35
+ return False
36
+
37
+ def main():
38
+ """Test different model options"""
39
+ print("🤖 Brello EI 0 - Model Options")
40
+ print("Created by Epic Systems | Engineered by Rehan Temkar")
41
+ print("=" * 60)
42
+
43
+ # Available model options
44
+ model_options = [
45
+ {
46
+ "path": "microsoft/DialoGPT-medium",
47
+ "description": "Public Model (Recommended for quick start)"
48
+ },
49
+ {
50
+ "path": "microsoft/DialoGPT-large",
51
+ "description": "Larger Public Model (Better responses)"
52
+ },
53
+ {
54
+ "path": "microsoft/DialoGPT-small",
55
+ "description": "Smaller Public Model (Faster)"
56
+ }
57
+ ]
58
+
59
+ # Test each model option
60
+ working_models = []
61
+ for option in model_options:
62
+ if test_model_option(option["path"], option["description"]):
63
+ working_models.append(option)
64
+
65
+ print(f"\n📊 Results:")
66
+ print(f"✅ Working models: {len(working_models)}/{len(model_options)}")
67
+
68
+ if working_models:
69
+ print("\n🎯 Recommended models:")
70
+ for model in working_models:
71
+ print(f" • {model['path']} - {model['description']}")
72
+
73
+ print("\n💡 To use Llama 3.2 3B:")
74
+ print("1. Create Hugging Face account")
75
+ print("2. Accept license at: https://huggingface.co/meta-llama/Llama-3.2-3B-Instruct")
76
+ print("3. Login with: huggingface-cli login")
77
+ print("4. Update model_path to: 'meta-llama/Llama-3.2-3B-Instruct'")
78
+
79
+ if __name__ == "__main__":
80
+ main()