raoufjat commited on
Commit
631b5f2
·
verified ·
1 Parent(s): 8596155

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -1,5 +1,3 @@
1
  # Load model directly
2
- from transformers import AutoTokenizer, AutoModelForCausalLM
3
-
4
- tokenizer = AutoTokenizer.from_pretrained("Ruqiya/Fine-Tuning-Gemma-2b-it-for-Arabic")
5
- model = AutoModelForCausalLM.from_pretrained("Ruqiya/Fine-Tuning-Gemma-2b-it-for-Arabic")
 
1
  # Load model directly
2
+ from transformers import AutoModel
3
+ model = AutoModel.from_pretrained("djmax13/qween7.5-arabic-story-teller-bnb-4bit")