Hugging Face's logo

Spaces:
abdelac
/
Mistral_Test
Runtime error

App Files Community
eesfeg commited on
Commit
bd49845
·
1 Parent(s): 25c6c06

fjf

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
8
  import gradio as gr
9
 
10
  # =================== CONFIGURATION ===================
11
- MODEL_ID = "abdelac/tinyllama" # Changed back to TinyLlama for CPU
12
  USE_CPU = True # Force CPU mode
13
 
14
  # =================== SUPPRESS WARNINGS ===================
 
8
  import gradio as gr
9
 
10
  # =================== CONFIGURATION ===================
11
+ MODEL_ID = "abdelac/Mistral_Test" # Changed back to TinyLlama for CPU abdelac/tinyllama
12
  USE_CPU = True # Force CPU mode
13
 
14
  # =================== SUPPRESS WARNINGS ===================