Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -74,8 +74,8 @@ def response_gen(user_review):
|
|
74 |
prompt = emotion_prompts.get(emotion_label, "Neutral").format(review=user_review)
|
75 |
|
76 |
# Load a pre-trained text generation model (replace 'meta-llama/Llama-3.2-1B' with an available model)
|
77 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
78 |
-
model = AutoModelForCausalLM.from_pretrained("
|
79 |
inputs = tokenizer(prompt, return_tensors="pt") # Tokenize the prompt
|
80 |
outputs = model.generate(**inputs, max_new_tokens=100) # Generate a response
|
81 |
|
|
|
74 |
prompt = emotion_prompts.get(emotion_label, "Neutral").format(review=user_review)
|
75 |
|
76 |
# Load a pre-trained text generation model (replace 'meta-llama/Llama-3.2-1B' with an available model)
|
77 |
+
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B")
|
78 |
+
model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B")
|
79 |
inputs = tokenizer(prompt, return_tensors="pt") # Tokenize the prompt
|
80 |
outputs = model.generate(**inputs, max_new_tokens=100) # Generate a response
|
81 |
|