Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,12 +2,12 @@ import gradio as gr
|
|
| 2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 3 |
import torch
|
| 4 |
|
| 5 |
-
#
|
| 6 |
models = {
|
| 7 |
-
"CodeGen (Salesforce)": "Salesforce/codegen-2B-multi",
|
| 8 |
-
"
|
| 9 |
-
"
|
| 10 |
-
"
|
| 11 |
}
|
| 12 |
|
| 13 |
# ΨͺΨΩ
ΩΩ Ψ§ΩΩΩ
Ψ§Ψ°Ψ¬
|
|
|
|
| 2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 3 |
import torch
|
| 4 |
|
| 5 |
+
# ΩΩ
Ψ§Ψ°Ψ¬ Ω
ΩΨͺΩΨΨ© ΩΨͺΩΩΩΨ― Ψ§ΩΩΩΨ―
|
| 6 |
models = {
|
| 7 |
+
"CodeGen 2B (Salesforce)": "Salesforce/codegen-2B-multi",
|
| 8 |
+
"WizardCoder 1B": "WizardLM/WizardCoder-1B-V1.0",
|
| 9 |
+
"CodeParrot Small": "codeparrot/codeparrot-small",
|
| 10 |
+
"GPT-J-6B (Python)": "EleutherAI/gpt-j-6B" # Ψ¨Ψ―ΩΩ Ω
ΩΨͺΩΨ ΩΩ Phind LLaMA
|
| 11 |
}
|
| 12 |
|
| 13 |
# ΨͺΨΩ
ΩΩ Ψ§ΩΩΩ
Ψ§Ψ°Ψ¬
|