Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ async def load_model():
|
|
45 |
try:
|
46 |
generator = pipeline(
|
47 |
'text-generation',
|
48 |
-
model='
|
49 |
device=device,
|
50 |
torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
|
51 |
)
|
|
|
45 |
try:
|
46 |
generator = pipeline(
|
47 |
'text-generation',
|
48 |
+
model='microsoft/Phi-3-mini-4k-instruct',
|
49 |
device=device,
|
50 |
torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
|
51 |
)
|