Hugging Face's logo

Spaces:
prithivMLmods
/
Callisto-OCR3-2B
Running on Zero

App Files Community
prithivMLmods commited on
Commit
ff334f0
·
verified ·
1 Parent(s): 2854d87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
25
  # ----------------------------------------------------------------------
26
  # 1. Setup Model & Tokenizer
27
  # ----------------------------------------------------------------------
28
- model_name = 'prithivMLmods/Raptor-X4' # Change as needed
29
  use_thread = True # Generation happens in a background thread
30
 
31
  model = AutoModelForCausalLM.from_pretrained(
 
25
  # ----------------------------------------------------------------------
26
  # 1. Setup Model & Tokenizer
27
  # ----------------------------------------------------------------------
28
+ model_name = 'prithivMLmods/Raptor-X6' # Change as needed
29
  use_thread = True # Generation happens in a background thread
30
 
31
  model = AutoModelForCausalLM.from_pretrained(