Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import pandas as pd
|
|
4 |
import os
|
5 |
import torch
|
6 |
import torch.nn as nn
|
7 |
-
from transformers import AutoTokenizer, AutoModelWithLMHead
|
8 |
from transformers.activations import get_activation
|
9 |
|
10 |
|
|
|
4 |
import os
|
5 |
import torch
|
6 |
import torch.nn as nn
|
7 |
+
from transformers import AutoTokenizer, AutoModelWithLMHead, AutoModelForCausalLM
|
8 |
from transformers.activations import get_activation
|
9 |
|
10 |
|