LovnishVerma commited on
Commit
259bc45
·
verified ·
1 Parent(s): d74c65e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -1,3 +1,10 @@
 
 
 
 
 
 
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
3
  import torch
 
1
+ # ⚠️ Performance Warning
2
+ # Running LLaMA 2 7B on CPU will be very slow. If performance is an issue, consider:
3
+
4
+ # Using a smaller model (e.g., llama-2-7b-chat-hf or mistral-7b).
5
+ # Upgrading to a GPU-enabled Hugging Face Space. (paid)
6
+
7
+
8
  import gradio as gr
9
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
10
  import torch