paulaamb commited on
Commit
368243d
·
verified ·
1 Parent(s): 512eb10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,8 +1,11 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
- import platform, os
5
- print(platform.processor())
 
 
 
6
 
7
  # Pick any model you want to test (small for free Spaces)
8
  MODEL_ID = "LiquidAI/LFM2-1.2B"
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
+ import os
5
+
6
+ # Log CPU info at startup
7
+ cpu_info = os.popen("cat /proc/cpuinfo | grep 'model name' | head -1").read().strip()
8
+ print(f"🖥️ Running on: {cpu_info}")
9
 
10
  # Pick any model you want to test (small for free Spaces)
11
  MODEL_ID = "LiquidAI/LFM2-1.2B"