Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|