Update app.py

app.py

@@ -31,23 +31,12 @@ Respond only with the category name, without any additional explanation or text.
 <|start_header_id|>assistant<|end_header_id|>
 '''
     
-    print("Sending the following prompt to the API:")
-    print(prompt)
-    
     output = query({"inputs": prompt})
     
-    print("\nAPI Response:")
-    print(output)
-    
     if isinstance(output, list) and len(output) > 0:
         generated_text = output[0].get('generated_text', '')
-        print("\nGenerated Text:")
-        print(generated_text)
-        # Extract the last non-empty line as the category
-        lines = [line.strip().lower() for line in generated_text.split('\n') if line.strip()]
-        if lines:
-            return lines[-1]
-    return "unknown"
+        return generated_text.strip()
+    return "Error: No response from the model"
 
 # Fetch a single post
 url = 'https://huggingface.co/posts'
@@ -58,11 +47,9 @@ if response.status_code == 200:
     pl7_element = soup.find(class_='pl-7')
     if pl7_element:
         pl7_text = pl7_element.text.strip()
-        print("Post content (first 100 characters):")
-        print(pl7_text[:100] + "..." if len(pl7_text) > 100 else pl7_text)
-        print("\nAnalyzing post...")
-        sentiment = analyze_sentiment(pl7_text)
-        print(f"\nSentiment category: {sentiment}")
+        print("Analyzing post...")
+        llm_response = analyze_sentiment(pl7_text)
+        print(f"\nLLM Response:\n{llm_response}")
     else:
         print("No post found with class 'pl-7'")
 else: