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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -1,8 +1,5 @@
1
- # Use a pipeline as a high-level helper
2
- from transformers import pipeline
3
 
4
- messages = [
5
- {"role": "user", "content": "Who are you?"},
6
- ]
7
- pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
8
- pipe(messages)
 
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")