mohamedemam commited on
Commit
2b6ad90
1 Parent(s): 78119f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -1,6 +1,19 @@
1
  from transformers import BitsAndBytesConfig
2
  import torch
3
-
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  config = PeftConfig.from_pretrained("mohamedemam/Arabic-meeting-summarization")
5
  model = AutoModelForCausalLM.from_pretrained("bigscience/bloomz-3b")
6
  model = PeftModel.from_pretrained(model, "mohamedemam/Arabic-meeting-summarization")
 
1
  from transformers import BitsAndBytesConfig
2
  import torch
3
+ import os
4
+ import torch
5
+ from datasets import load_dataset
6
+ from transformers import (
7
+ AutoModelForCausalLM,
8
+ AutoTokenizer,
9
+ BitsAndBytesConfig,
10
+ HfArgumentParser,
11
+ TrainingArguments,
12
+ pipeline,
13
+ logging,
14
+ Trainer
15
+ )
16
+ from peft import LoraConfig, PeftModel
17
  config = PeftConfig.from_pretrained("mohamedemam/Arabic-meeting-summarization")
18
  model = AutoModelForCausalLM.from_pretrained("bigscience/bloomz-3b")
19
  model = PeftModel.from_pretrained(model, "mohamedemam/Arabic-meeting-summarization")