Update app.py

app.py

@@ -2,18 +2,19 @@ from fastapi import FastAPI
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-# Skapa FastAPI-instans
 app = FastAPI()
 
-# Ladda GPT-SW3-126M
 model_name = "AI-Sweden-Models/gpt-sw3-126m"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+try:
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+except Exception as e:
+    print(f"Error loading model: {str(e)}")
+    raise e
 
-# Endpoint för att parsa aritmetiska prompts
 @app.post("/parse")
 async def parse_arithmetic_prompt(prompt: str):
-    if not prompt or len(prompt) > 100:  # Begränsa promptlängd
+    if not prompt or len(prompt) > 100:
         return {"error": "Ogiltig eller för lång prompt"}
     try:
         messages = [
@@ -28,7 +29,6 @@ async def parse_arithmetic_prompt(prompt: str):
     except Exception as e:
         return {"error": f"Fel vid parsning: {str(e)}"}
 
-# Hälsosida för att verifiera att Space körs
 @app.get("/")
 async def root():
     return {"message": "Arithmetic Parser API is running!"}