tahreem004 commited on
Commit
94ad44a
Β·
verified Β·
1 Parent(s): 8629951

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,9 +1,12 @@
1
- # Force install torch first before autoawq builds
2
  import os
3
  os.system("pip install torch")
4
 
5
- import torch # Now safe to import
 
6
 
 
 
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
  import gradio as gr
9
 
 
1
+ # Step 1: Force install torch first (if not already installed)
2
  import os
3
  os.system("pip install torch")
4
 
5
+ # Step 2: Install autoawq after torch is available
6
+ os.system("pip install autoawq")
7
 
8
+ # Now import everything safely
9
+ import torch
10
  from transformers import AutoTokenizer, AutoModelForCausalLM
11
  import gradio as gr
12