Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,15 +2,6 @@ import gradio as gr
|
|
2 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
3 |
import torch
|
4 |
import requests
|
5 |
-
import subprocess
|
6 |
-
|
7 |
-
# Ensure flash-attn is installed before using it
|
8 |
-
try:
|
9 |
-
import flash_attn
|
10 |
-
except ImportError:
|
11 |
-
print("Installing flash-attn...")
|
12 |
-
subprocess.run(["pip", "install", "flash-attn"], check=True)
|
13 |
-
import flash_attn
|
14 |
|
15 |
# Load DeepSeek-R1 model
|
16 |
model_name = "deepseek-ai/DeepSeek-R1"
|
|
|
2 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
3 |
import torch
|
4 |
import requests
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
|
6 |
# Load DeepSeek-R1 model
|
7 |
model_name = "deepseek-ai/DeepSeek-R1"
|