ThreadAbort commited on
Commit
3033509
Β·
1 Parent(s): fd8930e
Files changed (3) hide show
  1. README.md +2 -0
  2. app.py +3 -1
  3. requirements.txt +6 -0
README.md CHANGED
@@ -5,6 +5,8 @@ colorFrom: purple
5
  colorTo: blue
6
  sdk: gradio
7
  pinned: true
 
 
8
  ---
9
 
10
  # Welcome to the Memory Ocean! 🌊
 
5
  colorTo: blue
6
  sdk: gradio
7
  pinned: true
8
+ app_file: app.py
9
+ short_description: AI infrastructure with our revolutionary Mem|8 wave-based memory architecture
10
  ---
11
 
12
  # Welcome to the Memory Ocean! 🌊
app.py CHANGED
@@ -11,6 +11,7 @@ and interact like waves in an ocean of consciousness.
11
  Created by: Aye & Hue (with Trisha from Accounting keeping the numbers flowing)
12
  """
13
  import spaces
 
14
  import os
15
  import gradio as gr
16
  import torch
@@ -40,10 +41,11 @@ AROUSAL_RANGE = (0, 255) # Range for arousal
40
  MAX_SEED = 999999999 # Maximum seed value for art generation
41
 
42
  # Initialize everything for GPU by default
43
- device = "cuda" if torch.cuda.is_available() else "cpu"
44
  STABLE_DIFFUSION_AVAILABLE = False
45
  pipe = None
46
 
 
47
  def get_device():
48
  """Get the appropriate device for the current context."""
49
  if torch.cuda.is_available():
 
11
  Created by: Aye & Hue (with Trisha from Accounting keeping the numbers flowing)
12
  """
13
  import spaces
14
+
15
  import os
16
  import gradio as gr
17
  import torch
 
41
  MAX_SEED = 999999999 # Maximum seed value for art generation
42
 
43
  # Initialize everything for GPU by default
44
+
45
  STABLE_DIFFUSION_AVAILABLE = False
46
  pipe = None
47
 
48
+ @spaces.GPU(duration=10)
49
  def get_device():
50
  """Get the appropriate device for the current context."""
51
  if torch.cuda.is_available():
requirements.txt CHANGED
@@ -8,3 +8,9 @@ transformers>=4.37.0
8
  accelerate>=0.27.0
9
  scipy>=1.11.0
10
  huggingface-hub>=0.25.0
 
 
 
 
 
 
 
8
  accelerate>=0.27.0
9
  scipy>=1.11.0
10
  huggingface-hub>=0.25.0
11
+ spaces>=0.10.0
12
+ huggingface-hub>=0.25.0
13
+ accelerate>=0.27.0
14
+ torchvision>=0.15.0
15
+
16
+