added torch
Browse files- app.py +1 -1
- requirements.txt +2 -0
app.py
CHANGED
|
@@ -39,7 +39,7 @@ if uploaded_img is not None:
|
|
| 39 |
|
| 40 |
|
| 41 |
|
| 42 |
-
text_generator = pipeline("text-generation", model="meta-llama/Llama-2-7b-hf")
|
| 43 |
|
| 44 |
prompt = character_prompts[character]
|
| 45 |
st.write("Styled Prompt:", prompt)
|
|
|
|
| 39 |
|
| 40 |
|
| 41 |
|
| 42 |
+
text_generator = pipeline("text-generation", model="meta-llama/Llama-2-7b-hf", framework="pt")
|
| 43 |
|
| 44 |
prompt = character_prompts[character]
|
| 45 |
st.write("Styled Prompt:", prompt)
|
requirements.txt
CHANGED
|
@@ -4,3 +4,5 @@ tf-keras
|
|
| 4 |
Pillow
|
| 5 |
requests
|
| 6 |
tensorflow>=2.0
|
|
|
|
|
|
|
|
|
| 4 |
Pillow
|
| 5 |
requests
|
| 6 |
tensorflow>=2.0
|
| 7 |
+
torch
|
| 8 |
+
|