noddysnots commited on
Commit
d8fa37b
Β·
verified Β·
1 Parent(s): 3a51c6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -9
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"