ikraamkb commited on
Commit
9e440ae
·
verified ·
1 Parent(s): cdb0f41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ print(f"🔄 Loading model: {model_name}...")
25
  tokenizer = AutoTokenizer.from_pretrained(model_name)
26
  model = AutoModelForCausalLM.from_pretrained(model_name)
27
 
28
- qa_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=device)
29
 
30
  # Load Pretrained Object Detection Model (Torchvision)
31
  from torchvision.models.detection import FasterRCNN_ResNet50_FPN_Weights
 
25
  tokenizer = AutoTokenizer.from_pretrained(model_name)
26
  model = AutoModelForCausalLM.from_pretrained(model_name)
27
 
28
+ qa_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=-1)
29
 
30
  # Load Pretrained Object Detection Model (Torchvision)
31
  from torchvision.models.detection import FasterRCNN_ResNet50_FPN_Weights