Spaces:
Running
Running
Update app.py
Browse files
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
|
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
|